(*, args, checkpoint_map_location)
| 928 | |
| 929 | |
| 930 | def prior(*, args, checkpoint_map_location): |
| 931 | print("loading prior") |
| 932 | |
| 933 | prior_checkpoint = torch.load(args.prior_checkpoint_path, map_location=checkpoint_map_location) |
| 934 | |
| 935 | prior_model = prior_model_from_original_config() |
| 936 | |
| 937 | prior_diffusers_checkpoint = prior_original_checkpoint_to_diffusers_checkpoint(prior_model, prior_checkpoint) |
| 938 | |
| 939 | del prior_checkpoint |
| 940 | |
| 941 | load_prior_checkpoint_to_model(prior_diffusers_checkpoint, prior_model) |
| 942 | |
| 943 | print("done loading prior") |
| 944 | |
| 945 | return prior_model |
| 946 | |
| 947 | |
| 948 | def prior_image(*, args, checkpoint_map_location): |
no test coverage detected