(config, sd)
| 215 | |
| 216 | |
| 217 | def load_model_from_config(config, sd): |
| 218 | model = instantiate_from_config(config) |
| 219 | model.load_state_dict(sd,strict=False) |
| 220 | model.cuda() |
| 221 | model.eval() |
| 222 | return model |
| 223 | |
| 224 | |
| 225 | def load_model(config, ckpt, gpu, eval_mode): |
no test coverage detected