Merge pull request #55 from d8ahazard/main

Use shared models path
2024-11-04 15:05:07 +09:00 · 2024-11-04 15:05:07 +09:00 · d38bfc3c97
parent ff25bbf3ee 7741d8c08a
commit d38bfc3c97
1 changed files with 7 additions and 2 deletions
--- a/scripts/tools.py
+++ b/scripts/tools.py
@ -21,6 +21,11 @@ from collections import OrderedDict
 from PIL import Image

 model_cache = OrderedDict()
+models_path = shared.models_path
+sams_dir = os.path.join(models_path, "sam")
+if os.path.exists(sams_dir):
+    sam_model_dir = sams_dir
+else:
    sam_model_dir = os.path.join(
        extensions_dir, "PBRemTools/models/")
 model_list = [f for f in os.listdir(sam_model_dir) if os.path.isfile(