lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-14 17:37:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update preprocessor.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 20:48:31 -08:00
parent 13246270fa
commit 6e60e4696f
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extensions-builtin/forge_legacy_preprocessors/legacy_preprocessors/preprocessor.py
View File

@@ -711,8 +711,8 @@ class InsightFaceModel:
"""insightface's github release on antelopev2 model is down. Downloading
from huggingface mirror."""
from modules.modelloader import load_file_from_url
from annotator.annotator_path import models_path
model_root = os.path.join(models_path, "insightface", "models", "antelopev2")
from modules_forge.shared import models_path
model_root = os.path.join(models_path, "insightface", 'models', "antelopev2")
if not model_root:
os.makedirs(model_root, exist_ok=True)
for local_file, url in (
@@ -730,11 +730,12 @@ class InsightFaceModel:
def load_model(self):
if self.model is None:
from insightface.app import FaceAnalysis
from annotator.annotator_path import models_path
from modules_forge.shared import models_path
model_root = os.path.join(models_path, "insightface", 'models', "antelopev2")
self.model = FaceAnalysis(
name=self.face_analysis_model_name,
providers=['CUDAExecutionProvider', 'CPUExecutionProvider'],
root=os.path.join(models_path, "insightface"),
root=model_root,
)
self.model.prepare(ctx_id=0, det_size=(640, 640))