mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin 2023-08-09 14:56:43 -07:00
parent 69f57c0454
commit 54ce9a58ca
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
entry.py
View File

@ -130,10 +130,14 @@ def denoiser(input, sigma, c):
return model.denoiser(model.model, input, sigma, c) return model.denoiser(model.model, input, sigma, c)
with torch.no_grad():
model.model.to(torch.float16).cuda() model.model.to(torch.float16).cuda()
model.denoiser.to(torch.float16).cuda() model.denoiser.to(torch.float16).cuda()
samples_z = sampler(denoiser, randn, cond=c, uc=uc) samples_z = sampler(denoiser, randn, cond=c, uc=uc)
model.model.cpu() model.model.cpu()
model.denoiser.cpu() model.denoiser.cpu()
torch.cuda.empty_cache()
torch.cuda.ipc_collect()
a = 0 a = 0