(*, args, checkpoint_map_location)
| 1300 | |
| 1301 | |
| 1302 | def movq(*, args, checkpoint_map_location): |
| 1303 | print("loading movq") |
| 1304 | |
| 1305 | movq_checkpoint = torch.load(args.movq_checkpoint_path, map_location=checkpoint_map_location) |
| 1306 | |
| 1307 | movq_model = movq_model_from_original_config() |
| 1308 | |
| 1309 | movq_diffusers_checkpoint = movq_original_checkpoint_to_diffusers_checkpoint(movq_model, movq_checkpoint) |
| 1310 | |
| 1311 | del movq_checkpoint |
| 1312 | |
| 1313 | load_checkpoint_to_model(movq_diffusers_checkpoint, movq_model, strict=True) |
| 1314 | |
| 1315 | print("done loading movq") |
| 1316 | |
| 1317 | return movq_model |
| 1318 | |
| 1319 | |
| 1320 | def load_checkpoint_to_model(checkpoint, model, strict=False): |
no test coverage detected