Update __init__.py

This commit is contained in:
Silversith 2023-04-19 11:26:38 +02:00
parent 0a11af628b
commit 161223b76b

View File

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