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 15:05:02 -07:00
parent 9993090813
commit c56e2c9fd1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
entry.py
View File

@ -145,12 +145,12 @@ with torch.no_grad():
torch.cuda.ipc_collect() torch.cuda.ipc_collect()
with torch.no_grad(): with torch.no_grad():
model.first_stage_model.to(torch.float16).cuda() model.first_stage_model.cuda()
samples_x = model.decode_first_stage(samples_z) samples_x = model.decode_first_stage(samples_z.float())
samples = torch.clamp((samples_x + 1.0) / 2.0, min=0.0, max=1.0) samples = torch.clamp((samples_x + 1.0) / 2.0, min=0.0, max=1.0)
model.first_stage_model.cpu() model.first_stage_model.cpu()
import cv2 import cv2
samples = einops.rearrange(samples, 'b c h w -> b h w c')[0] * 127.5 + 127.5 samples = einops.rearrange(samples, 'b c h w -> b h w c')[0] * 255.0
samples = samples.cpu().numpy().clip(0, 255).astype(np.uint8) samples = samples.cpu().numpy().clip(0, 255).astype(np.uint8)
cv2.imwrite('img.png', samples) cv2.imwrite('img.png', samples)

BIN
img.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 4.7 KiB