timer_update

app.py

@@ -19,10 +19,10 @@ DEFAULT_MODEL_ID = os.getenv(
     "HF_ZERO_GPU_VIDEO_MODEL",
     "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
 )
-DEFAULT_GPU_DURATION_SECONDS = int(os.getenv("HF_ZERO_GPU_DURATION_SECONDS", "600"))
-DEFAULT_FPS = int(os.getenv("HF_ZERO_GPU_FPS", "7"))
-DEFAULT_NUM_FRAMES = int(os.getenv("HF_ZERO_GPU_NUM_FRAMES", "25"))
-DEFAULT_INFERENCE_STEPS = int(os.getenv("HF_ZERO_GPU_INFERENCE_STEPS", "25"))
+DEFAULT_GPU_DURATION_SECONDS = max(60, min(int(os.getenv("HF_ZERO_GPU_DURATION_SECONDS", "120")), 120))
+DEFAULT_FPS = int(os.getenv("HF_ZERO_GPU_FPS", "6"))
+DEFAULT_NUM_FRAMES = int(os.getenv("HF_ZERO_GPU_NUM_FRAMES", "16"))
+DEFAULT_INFERENCE_STEPS = int(os.getenv("HF_ZERO_GPU_INFERENCE_STEPS", "12"))
 DEFAULT_MOTION_BUCKET_ID = int(os.getenv("HF_ZERO_GPU_MOTION_BUCKET_ID", "127"))
 DEFAULT_NOISE_AUG_STRENGTH = float(os.getenv("HF_ZERO_GPU_NOISE_AUG_STRENGTH", "0.02"))
 
@@ -78,7 +78,7 @@ def _render_clip(
     seed: int,
 ) -> None:
     image = Image.open(image_path).convert("RGB")
-    target_size = (1024, 576) if image.width >= image.height else (576, 1024)
+    target_size = (768, 432) if image.width >= image.height else (432, 768)
     image = ImageOps.fit(image, target_size, method=Image.Resampling.LANCZOS)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     result = pipe(
@@ -134,10 +134,10 @@ def render_package(
             pipe,
             image_path,
             video_path,
-            num_frames=max(8, int(num_frames)),
-            inference_steps=max(10, int(inference_steps)),
+            num_frames=max(8, min(int(num_frames), 16)),
+            inference_steps=max(10, min(int(inference_steps), 12)),
             motion_bucket_id=int(motion_bucket_id),
-            fps=max(4, int(fps)),
+            fps=max(4, min(int(fps), 6)),
             noise_aug_strength=float(noise_aug_strength),
             seed=render_seed,
         )