Update README.md

README.md

@@ -11,7 +11,7 @@ The model can be directly load from pretrained with mochi branch: https://github
 from diffusers import MochiPipeline
 from diffusers.utils import export_to_video
 
-pipe = MochiPipeline.from_pretrained(path)
+pipe = MochiPipeline.from_pretrained(path, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 prompt = "Close-up of a chameleon's eye, with its scaly skin changing color. Ultra high resolution 4k."
 frames = pipe(prompt, num_inference_steps=28, guidance_scale=3.5).frames[0]