huggingface/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-05-28 00:39:35 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

note: torch.zeros -> torch.empty (#13807)

Browse Source 
* note: torch.zeros -> torch.empty

* move to gotchas.
This commit is contained in:
Sayak Paul
2026-05-26 10:37:33 +05:30
committed by GitHub
parent 3bc3e525fc
commit fdc6310f67
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.ai/models.md
View File

@@ -172,3 +172,5 @@ Boolean gate. If `False` (default), calling that method raises `ValueError`. All
freqs_dtype = torch.float32 if (is_mps or is_npu) else torch.float64
```
See `transformer_flux.py`, `transformer_flux2.py`, `transformer_wan.py`, `unet_2d_condition.py` for reference usages. Never leave an unconditional `torch.float64` in the model.
6. **Using `torch.empty`.** - Do not use `torch.empty` to initialize parameters. Use `torch.zeros` or `torch.ones`, instead.