[CLEANUP]

servers/text_to_video/test.py

@@ -5,7 +5,7 @@
     EulerDiscreteScheduler,
     MotionAdapter,
 )
-from diffusers.utils import export_to_gif
+from diffusers.utils import export_to_gif, export_to_video
 from dotenv import load_dotenv
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
@@ -60,29 +60,29 @@ def text_to_video(
             beta_schedule="linear",
         )
 
-        # outputs = []
-        # for i in range(n):
-        #     output = pipe(
-        #         prompt=task,
-        #         guidance_scale=guidance_scale,
-        #         num_inference_steps=inference_steps,
-        #     )
-        #     outputs.append(output)
-        #     if output_type == ".gif":
-        #         out = export_to_gif([output], f"{output_path}_{i}.gif")
-        #     else:
-        #         out = export_to_video([output], f"{output_path}_{i}.mp4")
-        output = pipe(
-            prompt = task,
-            guidance_scale = guidance_scale,
-            num_inference_steps = inference_steps
-        )
-        output = export_to_gif(output.frames[0], output_path)
-        return output
+        outputs = []
+        for i in range(n):
+            output = pipe(
+                prompt=task,
+                guidance_scale=guidance_scale,
+                num_inference_steps=inference_steps,
+            )
+            outputs.append(output)
+            if output_type == ".gif":
+                out = export_to_gif([output], f"{output_path}_{i}.gif")
+            else:
+                out = export_to_video([output], f"{output_path}_{i}.mp4")
+        # output = pipe(
+        #     prompt = task,
+        #     guidance_scale = guidance_scale,
+        #     num_inference_steps = inference_steps
+        # )
+        # output = export_to_gif(output.frames[0], output_path)
+        return out
     except Exception as e:
         logger.error(f"Error: {e}")
         return None
 
 
-out = text_to_video(task="A girl in hijab studying in a library")
+out = text_to_video(task="A girl in hijab studying in a library", n=5)
 print(out)