From 8a9bca5e36cebe16bd91d73a8dfcb24d54f937ee Mon Sep 17 00:00:00 2001
From: Radionic <khpluslt@gmail.com>
Date: Fri, 8 Sep 2023 16:36:33 +0800
Subject: [PATCH] fix: sam folder not found

---
 __init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/__init__.py b/__init__.py
index 91c9991..ed12c54 100644
--- a/__init__.py
+++ b/__init__.py
@@ -9,6 +9,7 @@ import os
 import sys
 import importlib
 import subprocess
+from folder_paths import add_model_folder_path
 
 ag_path = os.path.join(os.path.dirname(__file__))
 
@@ -32,6 +33,13 @@ def append_to_sys_path(path):
     if path not in sys.path:
         sys.path.append(path)
 
+def create_sam_model_dir():
+    model_dir = os.path.join(ag_path, "../../models/sam")
+    if not os.path.isdir(model_dir):
+        os.makedirs(model_dir)
+    add_model_folder_path('sam', model_dir)
+create_sam_model_dir()
+
 paths = ['blender', 'sam', 'common']
 files = []