should be >=

hkchengrex · Dec 22, 2024 · 8624361 · 8624361
1 parent 7ab89c2
commit 8624361
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gradio_demo.py b/gradio_demo.py
@@ -54,7 +54,7 @@ def video_to_audio(video: gr.Video, prompt: str, negative_prompt: str, seed: int
                    cfg_strength: float, duration: float):
 
     rng = torch.Generator(device=device)
-    if seed > 0:
+    if seed >= 0:
         rng.manual_seed(seed)
     else:
         rng.seed()
@@ -91,7 +91,7 @@ def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int,
                   duration: float):
 
     rng = torch.Generator(device=device)
-    if seed > 0:
+    if seed >= 0:
         rng.manual_seed(seed)
     else:
         rng.seed()