Function load_model

modelscope/pipelines/cv/image_to_3d_pipeline.py:41–48

(cfg, ckpt, strict=True)

# Load Syncdreamer Model
def load_model(cfg, ckpt, strict=True):
    config = OmegaConf.load(cfg)
    model = instantiate_from_config(config.model)
    print(f'loading model from {ckpt} ...')
    ckpt = torch.load(ckpt, map_location='cpu', weights_only=True)
    model.load_state_dict(ckpt['state_dict'], strict=strict)
    model = model.cuda().eval()
    return model


# Prepare Syncdreamer Input