-- Process 6 terminated with the following error:
Traceback (most recent call last):
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/multiprocessing/spawn.py", line 76, in _wrap
fn(i, *args)
File "/home/jundu/VGen-main/tools/train/train_dreamvideo_entrance.py", line 145, in worker
white_feature, _, zero_y = clip_encoder(image=clip_encoder.white_image, text="")
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 344, in forward
xt, x = self.encode_with_transformer(tokens.to(self.device), **kwargs)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 353, in encode_with_transformer
x = self.text_transformer_forward(x, attn_mask=self.model.attn_mask)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 366, in text_transformer_forward
x = r(x, attn_mask=attn_mask)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/open_clip/transformer.py", line 263, in forward
x = q_x + self.ls_1(self.attention(q_x=self.ln_1(q_x), k_x=k_x, v_x=v_x, attn_mask=attn_mask))
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/open_clip/transformer.py", line 250, in attention
return self.attn(
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/activation.py", line 1275, in forward
attn_output, attn_output_weights = F.multi_head_attention_forward(
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/functional.py", line 5438, in multi_head_attention_forward
raise RuntimeError(f"The shape of the 2D attn_mask is {attn_mask.shape}, but should be {correct_2d_size}.")
RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1).
-- Process 6 terminated with the following error:
Traceback (most recent call last):
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/multiprocessing/spawn.py", line 76, in _wrap
fn(i, *args)
File "/home/jundu/VGen-main/tools/train/train_dreamvideo_entrance.py", line 145, in worker
white_feature, _, zero_y = clip_encoder(image=clip_encoder.white_image, text="")
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 344, in forward
xt, x = self.encode_with_transformer(tokens.to(self.device), **kwargs)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 353, in encode_with_transformer
x = self.text_transformer_forward(x, attn_mask=self.model.attn_mask)
File "/home/jundu/VGen-main/tools/modules/clip_embedder.py", line 366, in text_transformer_forward
x = r(x, attn_mask=attn_mask)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/open_clip/transformer.py", line 263, in forward
x = q_x + self.ls_1(self.attention(q_x=self.ln_1(q_x), k_x=k_x, v_x=v_x, attn_mask=attn_mask))
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/open_clip/transformer.py", line 250, in attention
return self.attn(
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/modules/activation.py", line 1275, in forward
attn_output, attn_output_weights = F.multi_head_attention_forward(
File "/home/jundu/anaconda3/envs/dl/lib/python3.9/site-packages/torch/nn/functional.py", line 5438, in multi_head_attention_forward
raise RuntimeError(f"The shape of the 2D attn_mask is {attn_mask.shape}, but should be {correct_2d_size}.")
RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1).