wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 12:20:16 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
Silversith 2023-04-19 11:43:04 +02:00
parent 161223b76b
commit 2ba4d8eaed
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_nodes/facerestore/__init__.py
View File

@ -120,7 +120,7 @@ class FaceRestoreWithModel:
device = model_management.get_torch_device() device = model_management.get_torch_device()
upscale_model.to(device) upscale_model.to(device)
if self.face_helper is None: if self.face_helper is None:
self.face_helper = FaceRestoreHelper(1, face_size=768, crop_ratio=(1, 1), det_model=facedetection, save_ext='png', use_parse=True, device=device) self.face_helper = FaceRestoreHelper(1, face_size=512, crop_ratio=(1, 1), det_model=facedetection, save_ext='png', use_parse=True, device=device)
image_np = 255. * image_itm.cpu().numpy().squeeze() image_np = 255. * image_itm.cpu().numpy().squeeze()