Spaces:

THUDM
/

CogVideo

Running

akhaliq HF staff commited on Aug 3, 2022

Commit

669af6e

•

1 Parent(s): 970bd80

fix error

Files changed (1) hide show

model.py CHANGED Viewed

@@ -1192,7 +1192,7 @@ class Model:
             video_raw_text=text,
             video_guidance_text='视频',
             image_text_suffix=' 高清摄影',
-            batch_size=self.args.batch_size
             image_prompt=image_prompt)
         if not only_first_stage:
             _, res = self.process_stage2(

             video_raw_text=text,
             video_guidance_text='视频',
             image_text_suffix=' 高清摄影',
+            batch_size=self.args.batch_size,
             image_prompt=image_prompt)
         if not only_first_stage:
             _, res = self.process_stage2(