default ddim steps = 10

Ganjiali · Nov 8, 2024 · 6fe29d0 · 6fe29d0
1 parent 3b3f6e9
commit 6fe29d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generate.py b/generate.py
@@ -191,7 +191,7 @@ def main(args):
         help="What framerate should be used to save the output?",
         default=20,
     )
-    parse.add_argument("--ddim-steps", type=int, help="How many DDIM steps?", default=50)
+    parse.add_argument("--ddim-steps", type=int, help="How many DDIM steps?", default=10)
 
     args = parse.parse_args()
     print("inference args:")