1.0.19 (#33)

Unlock to allow changing model.

modules/path.py

@@ -5,3 +5,35 @@ lorafile_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../mode
 temp_outputs_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../outputs/'))
 
 os.makedirs(temp_outputs_path, exist_ok=True)
+
+default_base_model_name = 'sd_xl_base_1.0_0.9vae.safetensors'
+default_refiner_model_name = 'sd_xl_refiner_1.0_0.9vae.safetensors'
+default_lora_name = 'sd_xl_offset_example-lora_1.0.safetensors'
+default_lora_weight = 0.5
+
+model_filenames = []
+lora_filenames = []
+
+
+def get_model_filenames(folder_path):
+    if not os.path.isdir(folder_path):
+        raise ValueError("Folder path is not a valid directory.")
+
+    filenames = []
+    for filename in os.listdir(folder_path):
+        if os.path.isfile(os.path.join(folder_path, filename)):
+            _, file_extension = os.path.splitext(filename)
+            if file_extension.lower() in ['.pth', '.ckpt', '.bin', '.safetensors']:
+                filenames.append(filename)
+
+    return filenames
+
+
+def update_all_model_names():
+    global model_filenames, lora_filenames
+    model_filenames = get_model_filenames(modelfile_path)
+    lora_filenames = get_model_filenames(lorafile_path)
+    return
+
+
+update_all_model_names()