1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

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

minor updates to dtype map docs.
This commit is contained in:
Sayak Paul
2025-04-09 08:38:17 +05:30
committed by GitHub
parent b924251dd8
commit f685981ed0

View File

@@ -105,7 +105,7 @@ import torch
pipe = HunyuanVideoPipeline.from_pretrained(
"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)
```