[Admin maintenance] Support new ZeroGPU hardware

app.py

@@ -58,9 +58,13 @@ def save_video(video_array, video_save_path, fps: int = 16):
     video = (video + 1.0) / 2.0
     video = (video * 255).to(torch.uint8).permute(0, 2, 3, 1)
 
-    torchvision.io.write_video(
-        video_save_path, video, fps=fps, video_codec="h264", options={"crf": "10"}
-    )
+    frames = video.numpy()
+    writer = imageio.get_writer(video_save_path, fps=fps, codec="libx264", quality=8)
+    try:
+        for frame in frames:
+            writer.append_data(frame)
+    finally:
+        writer.close()
 
 example_txt = [
     "An astronaut riding a horse.",

requirements.txt

@@ -1,6 +1,4 @@
 numpy==1.26.4
-torch==2.2.2
-torchvision==0.17.2
 diffusers==0.30.0
 transformers==4.44.0
 accelerate==0.34.1