akhaliq HF staff commited on
Commit
c673115
1 Parent(s): 0f81310

just image prompt

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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=self.image_prompt, image_size=160).repeat(batch_size, 1).unsqueeze(1)
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