fix path for model
							parent
							
								
									4874282642
								
							
						
					
					
						commit
						a840986159
					
				|  | @ -11,7 +11,6 @@ from modules.face_analyser import get_one_face | |||
| from modules.typing import Frame, Face | ||||
| from modules.utilities import ( | ||||
|     conditional_download, | ||||
|     resolve_relative_path, | ||||
|     is_image, | ||||
|     is_video, | ||||
| ) | ||||
|  | @ -21,9 +20,11 @@ THREAD_SEMAPHORE = threading.Semaphore() | |||
| THREAD_LOCK = threading.Lock() | ||||
| NAME = "DLC.FACE-ENHANCER" | ||||
| 
 | ||||
| abs_dir = os.path.dirname(os.path.abspath(__file__)) | ||||
| models_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(abs_dir))), 'models') | ||||
| 
 | ||||
| def pre_check() -> bool: | ||||
|     download_directory_path = resolve_relative_path("..\models") | ||||
|     download_directory_path = models_dir | ||||
|     conditional_download( | ||||
|         download_directory_path, | ||||
|         [ | ||||
|  | @ -47,11 +48,7 @@ def get_face_enhancer() -> Any: | |||
| 
 | ||||
|     with THREAD_LOCK: | ||||
|         if FACE_ENHANCER is None: | ||||
|             if os.name == "nt": | ||||
|                 model_path = resolve_relative_path("..\models\GFPGANv1.4.pth") | ||||
|                 # todo: set models path https://github.com/TencentARC/GFPGAN/issues/399 | ||||
|             else: | ||||
|                 model_path = resolve_relative_path("../models/GFPGANv1.4.pth") | ||||
|             model_path = os.path.join(models_dir, 'GFPGANv1.4.pth') | ||||
|             FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1)  # type: ignore[attr-defined] | ||||
|     return FACE_ENHANCER | ||||
| 
 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue