reverting back

app.py

@@ -210,7 +210,6 @@ def run_one(
     action_id: int,
     out_dir: str,
     guidance: Tuple[float,float] = (1.0, 3.0),
-    seed = 0
 ) -> str:
     os.makedirs(out_dir, exist_ok=True)
     sample = make_sample(init_image_path, keyframes_boxes, action_id)
@@ -233,7 +232,6 @@ def run_one(
         use_factor_guidance=g_use_factor_guidance,
         guidance=list(guidance),
         video_path2=None,
-        seed=seed
     )
     return f"{video_prefix}.mp4"
 
@@ -315,10 +313,9 @@ def ctrl_generate_from_video(
     vid_path,
     action_label: str,
     num_frames: int = 3,
-    seed = 0, 
-    session_id = None,
     guidance_min: float = 1.0,
-    guidance_max: float = 3.0,
+    guidance_max: float = 3.0, 
+    session_id = None
 ):
     
     if session_id is None:
@@ -358,7 +355,6 @@ def ctrl_generate_from_video(
         action_id=action_id,
         out_dir=out_dir,
         guidance=(guidance_min, guidance_max),
-        seed = seed
     )
 
     if not os.path.exists(mp4_path):
@@ -370,10 +366,10 @@ def ctrl_generate_from_video(
 def ctrl_generate_from_image(
     img_path,
     action_label: str,
-    seed = 0,
-    session_id = None,
+    num_frames: int = 3,
     guidance_min: float = 1.0,
     guidance_max: float = 3.0, 
+    session_id = None
 ):
     
     if session_id is None:
@@ -421,7 +417,6 @@ def ctrl_generate_from_image(
         action_id=action_id,
         out_dir=out_dir,
         guidance=(guidance_min, guidance_max),
-        seed = seed
     )
 
     if not os.path.exists(mp4_path):
@@ -615,7 +610,7 @@ with gr.Blocks(css=css) as demo:
     run_video_btn.click(
         fn=ctrl_generate_from_video,
         inputs=[
-            video_in, action_dropdown, bbx_frames, 2, session_state
+            video_in, action_dropdown, bbx_frames
         ],
         outputs=[video_out, boxes_gif],
         api_name="generate"
@@ -624,7 +619,7 @@ with gr.Blocks(css=css) as demo:
     run_image_btn.click(
         fn=ctrl_generate_from_image,
         inputs=[
-            image_in, action_dropdown, 2, session_state
+            image_in, action_dropdown, bbx_frames
         ],
         outputs=[video_out],
         api_name="generate"

src/eval/generate_samples.py

@@ -144,7 +144,7 @@ def load_ctrlv_pipelines(model_dir, use_null_model=False, use_factor_guidance=Fa
     return pipeline
 
 
-def generate_video_ctrlv(sample, pipeline, video_path="video_out/genvid", json_path="video_out/gt_frames", bbox_mask_frames=None, action_type=None, use_factor_guidance=False, guidance=[1.0, 3.0], video_path2=None, seed=0):
+def generate_video_ctrlv(sample, pipeline, video_path="video_out/genvid", json_path="video_out/gt_frames", bbox_mask_frames=None, action_type=None, use_factor_guidance=False, guidance=[1.0, 3.0], video_path2=None):
     frame_size = (512, 320)
     FPS = 6
     CLIP_LENGTH = sample['bbox_images'].shape[0]
@@ -161,7 +161,7 @@ def generate_video_ctrlv(sample, pipeline, video_path="video_out/genvid", json_p
         json.dump(gt_frame_paths, file, indent=1)
     print("Saved GT frames json file:", json_path)
 
-    generator = torch.Generator(device=device).manual_seed(seed)
+    generator = torch.Generator(device=device).manual_seed(2)
 
     if not use_factor_guidance:
         frames = pipeline(init_image,