From 79282bfb71798047c0a424e562740d0f04a60ebd Mon Sep 17 00:00:00 2001
From: Charles Frye <charles@modal.com>
Date: Wed, 23 Oct 2024 21:10:20 -0700
Subject: [PATCH] boost num_inference_steps for better outputs

---
 06_gpu_and_ml/text-to-video/mochi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/06_gpu_and_ml/text-to-video/mochi.py b/06_gpu_and_ml/text-to-video/mochi.py
index 46a297676..3d4270038 100644
--- 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