From f08d17bccd94b6efbc81dbd6eda9445f8457d222 Mon Sep 17 00:00:00 2001 From: lobst Date: Thu, 14 Nov 2024 00:32:26 -0500 Subject: [PATCH] Enabled remote VitMatte Model download --- py/imagefunc.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/py/imagefunc.py b/py/imagefunc.py index a91bbc4..30d1fa8 100644 --- a/py/imagefunc.py +++ b/py/imagefunc.py @@ -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)