nodes/chflame163-ComfyUI_LayerStyle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #404 from evanshi80/main

Browse Source 
Enabled remote VitMatte Model download
This commit is contained in:
chflame163
2024-11-16 17:50:30 +08:00
committed by GitHub
parent 5f8b42724d f08d17bccd
commit 7ad0e976f6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/imagefunc.py
+3 -3
View File
@@ -1539,9 +1539,9 @@ class VITMatteModel:
self.processor = processor
def load_VITMatte_model(model_name:str, local_files_only:bool=False) -> object:
# if local_files_only:
# model_name = Path(os.path.join(folder_paths.models_dir, "vitmatte"))
model_name = Path(os.path.join(folder_paths.models_dir, "vitmatte"))
if local_files_only:
model_name = Path(os.path.join(folder_paths.models_dir, "vitmatte"))
# model_name = Path(os.path.join(folder_paths.models_dir, "vitmatte"))
from transformers import VitMatteImageProcessor, VitMatteForImageMatting
model = VitMatteForImageMatting.from_pretrained(model_name, local_files_only=local_files_only)
processor = VitMatteImageProcessor.from_pretrained(model_name, local_files_only=local_files_only)