diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index 29f5f88e..296c4192 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -41,12 +41,15 @@ def load_default_models(context: Context): for model_type in MODELS_TO_LOAD_ON_START: context.model_paths[model_type] = resolve_model_to_use(model_type=model_type) try: - load_model(context, model_type) + load_model( + context, + model_type, + scan_model = context.model_paths[model_type] != None and not context.model_paths[model_type].endswith('.safetensors') + ) except Exception as e: log.error(f"[red]Error while loading {model_type} model: {context.model_paths[model_type]}[/red]") log.exception(e) - def unload_all(context: Context): for model_type in KNOWN_MODEL_TYPES: unload_model(context, model_type)