|
|
|
@ -1,14 +1,13 @@
|
|
|
|
|
import os |
|
|
|
|
import time |
|
|
|
|
|
|
|
|
|
supported_ckpt_extensions = set(['.ckpt', '.pth', '.safetensors']) |
|
|
|
|
supported_pt_extensions = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors']) |
|
|
|
|
|
|
|
|
|
folder_names_and_paths = {} |
|
|
|
|
|
|
|
|
|
base_path = os.path.dirname(os.path.realpath(__file__)) |
|
|
|
|
models_dir = os.path.join(base_path, "models") |
|
|
|
|
folder_names_and_paths["checkpoints"] = ([os.path.join(models_dir, "checkpoints")], supported_ckpt_extensions) |
|
|
|
|
folder_names_and_paths["checkpoints"] = ([os.path.join(models_dir, "checkpoints")], supported_pt_extensions) |
|
|
|
|
folder_names_and_paths["configs"] = ([os.path.join(models_dir, "configs")], [".yaml"]) |
|
|
|
|
|
|
|
|
|
folder_names_and_paths["loras"] = ([os.path.join(models_dir, "loras")], supported_pt_extensions) |
|
|
|
|