45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
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/'))
|
|
vae_approx_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/vae_approx/'))
|
|
upscale_models_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/upscale_models/'))
|
|
temp_outputs_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../outputs/'))
|
|
|
|
fooocus_expansion_path = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
|
'../models/prompt_expansion/fooocus_expansion'))
|
|
|
|
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()
|