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

support klein edit

Browse Source
This commit is contained in:
Artiprocher
2026-01-20 12:58:18 +08:00
parent 88497b5c13
commit a835df984c
18 changed files with 273 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/flux2/model_inference_low_vram/FLUX.2-klein-4B.py
View File

@@ -25,3 +25,7 @@ pipe = Flux2ImagePipeline.from_pretrained(
prompt = "Masterpiece, best quality. Anime-style portrait of a woman in a blue dress, underwater, surrounded by colorful bubbles."
image = pipe(prompt, seed=0, rand_device="cuda", num_inference_steps=4)
image.save("image_FLUX.2-klein-4B.jpg")
prompt = "change the color of the clothes to red"
image = pipe(prompt, edit_image=[image], seed=1, rand_device="cuda", num_inference_steps=4)
image.save("image_edit_FLUX.2-klein-4B.jpg")

4
examples/flux2/model_inference_low_vram/FLUX.2-klein-9B.py
View File

@@ -25,3 +25,7 @@ pipe = Flux2ImagePipeline.from_pretrained(
prompt = "Masterpiece, best quality. Anime-style portrait of a woman in a blue dress, underwater, surrounded by colorful bubbles."
image = pipe(prompt, seed=0, rand_device="cuda", num_inference_steps=4)
image.save("image_FLUX.2-klein-9B.jpg")
prompt = "change the color of the clothes to red"
image = pipe(prompt, edit_image=[image], seed=1, rand_device="cuda", num_inference_steps=4)
image.save("image_edit_FLUX.2-klein-9B.jpg")

4
examples/flux2/model_inference_low_vram/FLUX.2-klein-base-4B.py
View File

@@ -25,3 +25,7 @@ pipe = Flux2ImagePipeline.from_pretrained(
prompt = "Masterpiece, best quality. Anime-style portrait of a woman in a blue dress, underwater, surrounded by colorful bubbles."
image = pipe(prompt, seed=0, rand_device="cuda", num_inference_steps=50, cfg_scale=4)
image.save("image_FLUX.2-klein-base-4B.jpg")
prompt = "change the color of the clothes to red"
image = pipe(prompt, edit_image=[image], seed=1, rand_device="cuda", num_inference_steps=50, cfg_scale=4)
image.save("image_edit_FLUX.2-klein-base-4B.jpg")

4
examples/flux2/model_inference_low_vram/FLUX.2-klein-base-9B.py
View File

@@ -25,3 +25,7 @@ pipe = Flux2ImagePipeline.from_pretrained(
prompt = "Masterpiece, best quality. Anime-style portrait of a woman in a blue dress, underwater, surrounded by colorful bubbles."
image = pipe(prompt, seed=0, rand_device="cuda", num_inference_steps=50, cfg_scale=4)
image.save("image_FLUX.2-klein-base-9B.jpg")
prompt = "change the color of the clothes to red"
image = pipe(prompt, edit_image=[image], seed=1, rand_device="cuda", num_inference_steps=50, cfg_scale=4)
image.save("image_edit_FLUX.2-klein-base-9B.jpg")