THUDM

/

CogVideoX1.5-5B

like 28

Follow

Knowledge Engineering Group (KEG) & Data Mining at Tsinghua University 1.49k

i get this error when using the provided ocde:
"""
Trying to set a tensor of shape torch.Size([3072, 128]) in "weight" (which has shape torch.Size([3072, 16, 2, 2])), this look incorrect.
File "/home/kurama/Dokumente/vsProjects/vid_gen_5b_1.5/main_nq.py", line 7, in
pipe = CogVideoXPipeline.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: Trying to set a tensor of shape torch.Size([3072, 128]) in "weight" (which has shape torch.Size([3072, 16, 2, 2])), this look incorrect.
"""