| import torch | |
| from diffsynth import ModelManager, FluxImagePipeline, download_customized_models | |
| lora_path = download_customized_models( | |
| model_id="DiffSynth-Studio/ArtAug-lora-FLUX.1dev-v1", | |
| origin_file_path="merged_lora.safetensors", | |
| local_dir="models/lora" | |
| )[0] | |
| model_manager = ModelManager(torch_dtype=torch.bfloat16, device="cuda", model_id_list=["FLUX.1-dev"]) | |
| model_manager.load_lora(lora_path, lora_alpha=1.0) | |
| pipe = FluxImagePipeline.from_model_manager(model_manager) | |
| image = pipe(prompt="a house", seed=0) | |
| image.save("image_artaug.jpg") | |