Spaces:

rcfeng
/

KEEP

Running on Zero

rcfeng commited on 14 days ago

Commit

359740f

1 Parent(s): 3a4ff9c

fix parse

Files changed (1) hide show

facelib/utils/face_restoration_helper.py CHANGED Viewed

@@ -133,8 +133,9 @@ class FaceRestoreHelper(object):
         # init face parsing model
         self.use_parse = use_parse
-        self.face_parse = init_parsing_model(
-            model_name='parsenet', device=self.device)
     def set_upscale_factor(self, upscale_factor):
         self.upscale_factor = upscale_factor

         # init face parsing model
         self.use_parse = use_parse
+        if self.use_parse:
+            self.face_parse = init_parsing_model(
+                model_name='parsenet', device=self.device)
     def set_upscale_factor(self, upscale_factor):
         self.upscale_factor = upscale_factor