From e87eacf6edd6d4d74c5d2caf87b89b3b08cf1c02 Mon Sep 17 00:00:00 2001 From: Radionic Date: Tue, 3 Oct 2023 14:07:41 +0800 Subject: [PATCH] fix: sams folder path --- __init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/__init__.py b/__init__.py index 36d3483..50c5e80 100644 --- a/__init__.py +++ b/__init__.py @@ -15,6 +15,7 @@ import sys import importlib import subprocess import requests +import folder_paths from folder_paths import add_model_folder_path, get_filename_list, get_folder_paths from tqdm import tqdm @@ -42,6 +43,7 @@ def append_to_sys_path(path): if path not in sys.path: sys.path.append(path) +folder_paths.folder_names_and_paths["sams"] = ([os.path.join(folder_paths.models_dir, "sams")], folder_paths.supported_pt_extensions) def download_sam_model(): model_dir = get_folder_paths("sams")[0]