only load libs when necessary

This commit is contained in:
lllyasviel
2023-11-12 02:16:11 -08:00
parent e7fe1d443a
commit e6aeefd2b4
3 changed files with 4 additions and 5 deletions
+2 -2
View File
@@ -1,7 +1,6 @@
import cv2
import numpy as np
import modules.config
import facexlib.utils.face_restoration_helper as face_restoration_helper
faceRestoreHelper = None
@@ -26,7 +25,8 @@ def crop_image(img_rgb):
global faceRestoreHelper
if faceRestoreHelper is None:
faceRestoreHelper = face_restoration_helper.FaceRestoreHelper(
from facexlib.utils.face_restoration_helper import FaceRestoreHelper
faceRestoreHelper = FaceRestoreHelper(
upscale_factor=1,
model_rootpath=modules.config.path_controlnet,
device='cpu' # use cpu is safer since we are out of fcbh management