fix OOM for test_vae_tiling (#7510) - SemanticDiff

Commit

1 year ago

fix OOM for test_vae_tiling (#7510) use float16 and add torch.no_grad()

References

#7510 - fix OOM for test_vae_tiling

Author

yiyixuxu

yiyixuxu

Parents

Loading