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:00:23 -07:00
parent 54ce9a58ca
commit ab65a357ec
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
entry.py
+14
View File
@@ -1,5 +1,7 @@
import os import os
import math import math
import einops
import numpy as np import numpy as np
import torch import torch
import gc import gc
@@ -141,3 +143,15 @@ torch.cuda.empty_cache()
torch.cuda.ipc_collect() torch.cuda.ipc_collect()
a = 0 a = 0
with torch.no_grad():
model.first_stage_model.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()
import cv2
samples = einops.rearrange(samples, 'b c h w -> b h w c')[0] * 127.5 + 127.5
samples = samples.cpu().numpy().clip(0, 255).astype(np.uint8)
cv2.imwrite('img.png', samples)