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

Update preprocessor.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 20:49:26 -08:00
parent 6e60e4696f
commit 1a83cc522c
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -712,7 +712,7 @@ class InsightFaceModel:
from huggingface mirror."""
from modules.modelloader import load_file_from_url
from modules_forge.shared import models_path
model_root = os.path.join(models_path, "insightface", 'models', "antelopev2")
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 (
@@ -731,11 +731,10 @@ class InsightFaceModel:
if self.model is None:
from insightface.app import FaceAnalysis
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=model_root,
root=os.path.join(models_path, "insightface"),
)
self.model.prepare(ctx_id=0, det_size=(640, 640))