(*, args, checkpoint_map_location)
| 865 | |
| 866 | |
| 867 | def prior(*, args, checkpoint_map_location): |
| 868 | print("loading prior") |
| 869 | |
| 870 | prior_checkpoint = torch.load(args.prior_checkpoint_path, map_location=checkpoint_map_location) |
| 871 | |
| 872 | clip_stats_checkpoint = torch.load(args.clip_stat_path, map_location=checkpoint_map_location) |
| 873 | |
| 874 | prior_model = prior_model_from_original_config() |
| 875 | |
| 876 | prior_diffusers_checkpoint = prior_original_checkpoint_to_diffusers_checkpoint( |
| 877 | prior_model, prior_checkpoint, clip_stats_checkpoint |
| 878 | ) |
| 879 | |
| 880 | del prior_checkpoint |
| 881 | del clip_stats_checkpoint |
| 882 | |
| 883 | load_checkpoint_to_model(prior_diffusers_checkpoint, prior_model, strict=True) |
| 884 | |
| 885 | print("done loading prior") |
| 886 | |
| 887 | return prior_model |
| 888 | |
| 889 | |
| 890 | def text2img(*, args, checkpoint_map_location): |
no test coverage detected