From 2899283c017470183752c38a09caa27f91e272fb Mon Sep 17 00:00:00 2001
From: Zhongjie Duan <35051019+Artiprocher@users.noreply.github.com>
Date: Tue, 18 Feb 2025 18:00:08 +0800
Subject: [PATCH] Update stepvideo examples

---
 examples/stepvideo/stepvideo_text_to_video_low_vram.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/stepvideo/stepvideo_text_to_video_low_vram.py b/examples/stepvideo/stepvideo_text_to_video_low_vram.py
index 95d4557..f75ae16 100644
--- a/examples/stepvideo/stepvideo_text_to_video_low_vram.py
+++ b/examples/stepvideo/stepvideo_text_to_video_low_vram.py
@@ -28,7 +28,7 @@ model_manager.load_models(
             "models/stepfun-ai/stepvideo-t2v/transformer/diffusion_pytorch_model-00006-of-00006.safetensors",
         ]
     ],
-    torch_dtype=torch.float8_e4m3fn, device="cpu"
+    torch_dtype=torch.bfloat16, device="cpu" # You can set torch_dtype=torch.bfloat16 to reduce RAM (not VRAM) usage.
 )
 model_manager.load_models(
     ["models/stepfun-ai/stepvideo-t2v/vae/vae_v2.safetensors"],