24 lines
566 B
Python
24 lines
566 B
Python
import os
|
|
import folder_paths
|
|
|
|
import comfy.utils
|
|
|
|
from .. import constants
|
|
|
|
|
|
def get_model_dir(model_dir):
|
|
root_path = folder_paths.models_dir
|
|
path = os.path.join(root_path, constants.INSTANCE_MODELS_DIR, model_dir)
|
|
return path
|
|
|
|
|
|
def get_model_list(model_dir) -> list[str]:
|
|
path = get_model_dir(model_dir)
|
|
return os.listdir(path)
|
|
|
|
|
|
def load_checkpoint(model_dir, filename):
|
|
checkpoint_path = os.path.join(get_model_dir(model_dir), filename)
|
|
checkpoint = comfy.utils.load_torch_file(checkpoint_path, safe_load=True)
|
|
return checkpoint
|