boost num_inference_steps for better outputs

modal-labs · Oct 24, 2024 · 79282bf · 79282bf
1 parent 94b1095
commit 79282bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/06_gpu_and_ml/text-to-video/mochi.py b/06_gpu_and_ml/text-to-video/mochi.py
@@ -250,7 +250,7 @@ def generate_video(
         num_frames=163,
         seed=12345,
         cfg_scale=4.5,
-        num_inference_steps=64,
+        num_inference_steps=200,
     ):
         # credit: https://github.com/genmoai/models/blob/7c7d33c49d53bbf939fd6676610e949f3008b5a8/src/mochi_preview/infer.py#L63