open_clip_torch>=2.26.0版本的兼容性问题发现与解决方案

举例:在I2VGen-XL的推理环境安装中,如果安装了2.24.0以后版本的 open_clip_torch ,那么这些版本中的Transformer具有默认的 batch_first=True 参数,特别是在函数调用中很难找到如何在传递中增加这个变更后的参数。

这可能导致出现如下报错:

Exception: Failed to invoke function <function inference_i2vgen_entrance at 0x7f5e6f958c10>, with The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1).

解决方案其实可以直接参考open_clip_torch的变更方法,即:
commit

对于VGen来说,就是注释图中的两行对x的变换。源代码位于Github/VGen/tools/modules/clip_embedder.py
修订方法

  • 6
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值