(args)
| 80 | |
| 81 | |
| 82 | def main(args): |
| 83 | original_ckpt = load_original_checkpoint(args) |
| 84 | |
| 85 | num_layers = 19 |
| 86 | converted_ip_adapter_state_dict = convert_flux_ipadapter_checkpoint_to_diffusers(original_ckpt, num_layers) |
| 87 | |
| 88 | print("Saving Flux IP-Adapter in Diffusers format.") |
| 89 | safetensors.torch.save_file(converted_ip_adapter_state_dict, f"{args.output_path}/model.safetensors") |
| 90 | |
| 91 | if vision: |
| 92 | model = CLIPVisionModelWithProjection.from_pretrained(args.vision_pretrained_or_path) |
| 93 | model.save_pretrained(f"{args.output_path}/image_encoder") |
| 94 | |
| 95 | |
| 96 | if __name__ == "__main__": |
no test coverage detected
searching dependent graphs…