diff --git a/modules/sd_models.py b/modules/sd_models.py index dc0679c58..a925b207e 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -19,7 +19,7 @@ from modules import paths, shared, shared_state, modelloader, devices, script_ca from modules.timer import Timer from modules.memstats import memory_stats from modules.modeldata import model_data -from modules.sd_checkpoint import CheckpointInfo, select_checkpoint, list_models, checkpoints_list, checkpoint_titles, get_closet_checkpoint_match, update_model_hashes, setup_model, write_metadata, read_metadata_from_safetensors # pylint: disable=unused-import +from modules.sd_checkpoint import CheckpointInfo, select_checkpoint, list_models, checkpoints_list, checkpoint_titles, get_closet_checkpoint_match, model_hash, update_model_hashes, setup_model, write_metadata, read_metadata_from_safetensors # pylint: disable=unused-import model_dir = "Stable-diffusion"