(*, args, checkpoint_map_location)
| 906 | |
| 907 | |
| 908 | def inpaint_text2img(*, args, checkpoint_map_location): |
| 909 | print("loading inpaint text2img") |
| 910 | |
| 911 | inpaint_text2img_checkpoint = torch.load( |
| 912 | args.inpaint_text2img_checkpoint_path, map_location=checkpoint_map_location |
| 913 | ) |
| 914 | |
| 915 | inpaint_unet_model = inpaint_unet_model_from_original_config() |
| 916 | |
| 917 | inpaint_unet_diffusers_checkpoint = inpaint_unet_original_checkpoint_to_diffusers_checkpoint( |
| 918 | inpaint_unet_model, inpaint_text2img_checkpoint |
| 919 | ) |
| 920 | |
| 921 | del inpaint_text2img_checkpoint |
| 922 | |
| 923 | load_checkpoint_to_model(inpaint_unet_diffusers_checkpoint, inpaint_unet_model, strict=True) |
| 924 | |
| 925 | print("done loading inpaint text2img") |
| 926 | |
| 927 | return inpaint_unet_model |
| 928 | |
| 929 | |
| 930 | # movq |
no test coverage detected