theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update artaug_flux.py

Browse Source
This commit is contained in:
Zhongjie Duan
2024-12-18 20:43:53 +08:00
committed by GitHub
parent 95a0f0bedc
commit 29cebf0bec
1 changed files with 4 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
examples/ArtAug/artaug_flux.py
View File

@@ -1,24 +1,14 @@
import torch import torch
from diffsynth import ModelManager, FluxImagePipeline, download_customized_models from diffsynth import ModelManager, FluxImagePipeline, download_customized_models
prompt = "a beautiful Asian girl."
# Generate an image using FLUX.1-dev
model_manager = ModelManager(torch_dtype=torch.bfloat16, device="cuda", model_id_list=["FLUX.1-dev"])
pipe = FluxImagePipeline.from_model_manager(model_manager)
image = pipe(prompt=prompt, seed=0)
image.save("image.jpg")
# Download and load ArtAug LoRA
lora_path = download_customized_models( lora_path = download_customized_models(
model_id="DiffSynth-Studio/ArtAug-lora-FLUX.1dev-v1", model_id="DiffSynth-Studio/ArtAug-lora-FLUX.1dev-v1",
origin_file_path="merged_lora.safetensors", origin_file_path="merged_lora.safetensors",
local_dir="models/lora", local_dir="models/lora"
downloading_priority=["ModelScope", "HuggingFace"]
)[0] )[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) model_manager.load_lora(lora_path, lora_alpha=1.0)
pipe = FluxImagePipeline.from_model_manager(model_manager)
# Generate an image using FLUX.1-dev + ArtAug image = pipe(prompt="a house", seed=0)
image = pipe(prompt=prompt, seed=0)
image.save("image_artaug.jpg") image.save("image_artaug.jpg")