PR #7510 fix OOM for test_vae_tiling

fix OOM for test_vae_tiling #7510

sayakpaul merged 1 commit into main from fix-tilling-test

use float16 and add torch.no_grad()

8d853a92

yiyixuxu requested a review from

sayakpaul 2 years ago

tolgacangoz approved these changes on 2024-03-28

sayakpaul commented on 2024-03-29

sayakpaul approved these changes on 2024-03-29

sayakpaul merged 34c90dbb into main 2 years ago

sayakpaul deleted the fix-tilling-test branch 2 years ago

Reviewers

sayakpaul

tolgacangoz

Assignees

No one assigned

Labels

None yet

Milestone

No milestone