Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin
2023-08-09 15:02:24 -07:00
parent ab65a357ec
commit 9993090813
1 changed files with 50 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
entry.py
+3 -4
View File
@@ -98,6 +98,8 @@ model.load_state_dict(safetensors.torch.load_file('./sd_xl_base_1.0.safetensors'
# model.conditioner.cuda()
with torch.no_grad():
model.conditioner.embedders[0].device = 'cpu'
model.conditioner.embedders[1].device = 'cpu'
@@ -142,11 +144,8 @@ with torch.no_grad():
torch.cuda.empty_cache()
torch.cuda.ipc_collect()
a = 0
with torch.no_grad():
model.first_stage_model.cuda()
model.first_stage_model.to(torch.float16).cuda()
samples_x = model.decode_first_stage(samples_z)
samples = torch.clamp((samples_x + 1.0) / 2.0, min=0.0, max=1.0)
model.first_stage_model.cpu()