Function get_model

src/composenW.py:34–37

(path)

def get_model(path):
    config = OmegaConf.load("configs/custom-diffusion/finetune.yaml")
    model = load_model_from_config(config, path)
    return model, config


def gdupdateWexact(K, V, Ktarget1, Vtarget1, W, device='cuda'):