diff --git a/.gitignore b/.gitignore
index ee78b59..3554280 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,4 +4,4 @@ __pycache__/
 .DS_Store
 
 # settings
-models/*
\ No newline at end of file
+models/*.ckpt
\ No newline at end of file
diff --git a/scripts/stablesr.py b/scripts/stablesr.py
index 1689137..f98eb81 100644
--- a/scripts/stablesr.py
+++ b/scripts/stablesr.py
@@ -119,6 +119,8 @@ class Script(scripts.Script):
     def load_model_list(self):
         # traverse the CFG_PATH and add all files to the model list
         self.model_list = {}
+        if not MODEL_PATH.exists():
+            MODEL_PATH.mkdir()
         for file in MODEL_PATH.iterdir():
             if file.is_file():
                 # save tha absolute path