model path location fixed [errno 2] no such file or directory

Author: daxthin

DZFaceDetailer.py

@@ -9,7 +9,7 @@
 import nodes
 from folder_paths import base_path
 
-face_model_path = os.path.join(base_path, "models/DZ-FaceDetailer/yolo/face_yolov8n.pt")
+face_model_path = os.path.join(base_path, "models/dz_facedetailer/yolo/face_yolov8n.pt")
 MASK_CONTROL = ["dilate", "erode", "disabled"]
 MASK_TYPE = ["face", "box"]
 
@@ -51,15 +51,9 @@ def detailer(self, model, seed, steps, cfg, sampler_name, scheduler, positive, n
         img = image2nparray(input_tensor_img, False)
         # Process the face mesh or make the face box for masking
         if mask_type == "box":
-            try:
-                final_mask = facebox_mask(img)
-            except:
-                return (latent_image, )
+            final_mask = facebox_mask(img)
         else:
-            try:
-                final_mask = facemesh_mask(img)
-            except:
-                return (latent_image, )
+            final_mask = facemesh_mask(img)
         # Erode/Dilate mask
         if mask_control == "dilate":
             if dilate_mask_value > 0:

__init__.py

@@ -15,7 +15,7 @@
 
 model_url = "https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8n.pt"
 
-save_loc = os.path.join(models_dir, "DZ-FaceDetailer\yolo")
+save_loc = os.path.join(models_dir, "dz_facedetailer", "yolo")
 
 def download_model():
     if Path(os.path.join(save_loc, "face_yolov8n.pt")).is_file():
@@ -46,13 +46,13 @@ def download_model():
         except requests.exceptions.RequestException as err:
             print('FaceDetailer: Model download failed: {err}')
             print(f'FaceDetailer: Download it manually from: {model_url}')
-            print('FaceDetailer: And put it in /comfyui/models/DZ-FaceDetailer/yolo/')
+            print('FaceDetailer: And put it in /comfyui/models/dz_facedetailer/yolo/')
         except Exception as e:
             print(f'FaceDetailer: An unexpected error occurred: {e}')
 
 if not os.path.exists(save_loc):
     print('FaceDetailer: Creating models directory')
-    os.makedirs(save_loc)
+    os.makedirs(save_loc, exist_ok=True)
     download_model()
 else:
     print('FaceDetailer: Model directory already exists')