Spaces:
Running
Running
just image prompt
Browse files
model.py
CHANGED
@@ -858,7 +858,7 @@ class Model:
|
|
858 |
1
|
859 |
) # given_tokens.shape: [bs, frame_num, 400]
|
860 |
else:
|
861 |
-
given_tokens = tokenizer.encode(image_path=
|
862 |
|
863 |
# generate subsequent frames:
|
864 |
total_frames = generate_frame_num
|
|
|
858 |
1
|
859 |
) # given_tokens.shape: [bs, frame_num, 400]
|
860 |
else:
|
861 |
+
given_tokens = tokenizer.encode(image_path=image_prompt, image_size=160).repeat(batch_size, 1).unsqueeze(1)
|
862 |
|
863 |
# generate subsequent frames:
|
864 |
total_frames = generate_frame_num
|