Unverified Commit f685981e authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[docs] minor updates to dtype map docs. (#11237)

minor updates to dtype map docs.
parent b924251d
...@@ -105,7 +105,7 @@ import torch ...@@ -105,7 +105,7 @@ import torch
pipe = HunyuanVideoPipeline.from_pretrained( pipe = HunyuanVideoPipeline.from_pretrained(
"hunyuanvideo-community/HunyuanVideo", "hunyuanvideo-community/HunyuanVideo",
torch_dtype={'transformer': torch.bfloat16, 'default': torch.float16}, torch_dtype={"transformer": torch.bfloat16, "default": torch.float16},
) )
print(pipe.transformer.dtype, pipe.vae.dtype) # (torch.bfloat16, torch.float16) print(pipe.transformer.dtype, pipe.vae.dtype) # (torch.bfloat16, torch.float16)
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment