()
| 6 | |
| 7 | |
| 8 | def build_pipeline(): |
| 9 | model_manager = ModelManager(torch_dtype=torch.bfloat16, device="cuda", model_id_list=["FLUX.1-dev"]) |
| 10 | model_manager.load_lora( |
| 11 | download_customized_models( |
| 12 | model_id="DiffSynth-Studio/Eligen", |
| 13 | origin_file_path="model_bf16.safetensors", |
| 14 | local_dir="models/lora/entity_control" |
| 15 | ), |
| 16 | lora_alpha=1 |
| 17 | ) |
| 18 | model_manager.load_lora( |
| 19 | download_customized_models( |
| 20 | model_id="iic/In-Context-LoRA", |
| 21 | origin_file_path="visual-identity-design.safetensors", |
| 22 | local_dir="models/lora/In-Context-LoRA" |
| 23 | ), |
| 24 | lora_alpha=1 |
| 25 | ) |
| 26 | pipe = FluxImagePipeline.from_model_manager(model_manager) |
| 27 | return pipe |
| 28 | |
| 29 | |
| 30 | def generate(pipe: FluxImagePipeline, source_image, target_image, mask, height, width, prompt, entity_prompt, image_save_path, mask_save_path, seed=0): |
no test coverage detected