tests/models/autoencoders/test_models_vae.py · 34c90dbb31f4956e72086ac43ca7d8154c2aadae · renzhc / diffusers_dcu · GitLab

Find file Blame History Permalink

fix OOM for test_vae_tiling (#7510) · 34c90dbb
YiYi Xu authored Mar 28, 2024
```
use float16 and add torch.no_grad()
```
34c90dbb

test_models_vae.py 39.7 KB