up

import collections

import copy

import os.path

import sys

import gc

    if checkpoint_info in checkpoints_loaded:

        # use checkpoint cache

        print(f"Loading weights [{sd_model_hash}] from cache")

        # move to end as latest

        checkpoints_loaded.move_to_end(checkpoint_info)

        return checkpoints_loaded[checkpoint_info]

    print(f"Loading weights [{sd_model_hash}] from {checkpoint_info.filename}")

    if model.is_sdxl:

        sd_models_xl.extend_sdxl(model)

    model.load_state_dict(state_dict, strict=False)

    timer.record("apply weights to model")

    if shared.opts.sd_checkpoint_cache > 0:

        # cache newly loaded model

        checkpoints_loaded[checkpoint_info] = state_dict

        checkpoints_loaded[checkpoint_info] = copy.deepcopy(state_dict)

    model.load_state_dict(state_dict, strict=False)

    timer.record("apply weights to model")

    del state_dict