| import os |
|
|
| modelfile_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/checkpoints/')) |
| lorafile_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/loras/')) |
| 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() |
|
|
