theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-04-21 11:46:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

refine code

Browse Source
This commit is contained in:
Artiprocher
2026-04-20 14:31:26 +08:00
parent b51fac3e0e
commit 5e10e11dfc
2 changed files with 49 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
examples/flux2/model_inference/Template-KleinBase4B-Age.py
View File

@@ -17,15 +17,27 @@ template = TemplatePipeline.from_pretrained(
device="cuda",
model_configs=[ModelConfig(model_id="DiffSynth-Studio/Template-KleinBase4B-Age")],
)
prompt = "Half body color photograph of a single woman, head and torso with visible arms and hands resting gently in front of the body, looking directly at the camera, centered composition, colorful studio background with soft gradient of warm pastel tones, vibrant studio lighting, wearing a plain red short-sleeve t-shirt, straight black shoulder-length hair, photorealistic, high quality"# prompt = "Full body photograph of a single woman standing, looking directly at the camera, centered composition, plain neutral gray background, soft even studio lighting, wearing a plain white short-sleeve t-shirt and blue jeans, barefoot, arms resting naturally at sides, straight black shoulder-length hair, photorealistic, high quality"
negative_age = 45
for age in range(10, 91, 5):
print(f"Generating age {age}...")
image = template(
pipe,
prompt=prompt,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": age}],
negative_template_inputs=[{"age": negative_age}],
template_inputs=[{"age": 20}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_{age}.jpg")
image.save(f"image_age_20.jpg")
image = template(
pipe,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": 50}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_50.jpg")
image = template(
pipe,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": 80}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_80.jpg")

29
examples/flux2/model_inference_low_vram/Template-KleinBase4B-Age.py
View File

@@ -1,6 +1,7 @@
from diffsynth.diffusion.template import TemplatePipeline
from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig
import torch
vram_config = {
"offload_dtype": "disk",
"offload_device": "disk",
@@ -28,15 +29,27 @@ template = TemplatePipeline.from_pretrained(
model_configs=[ModelConfig(model_id="DiffSynth-Studio/Template-KleinBase4B-Age")],
lazy_loading=True,
)
prompt = "Half body color photograph of a single woman, head and torso with visible arms and hands resting gently in front of the body, looking directly at the camera, centered composition, colorful studio background with soft gradient of warm pastel tones, vibrant studio lighting, wearing a plain red short-sleeve t-shirt, straight black shoulder-length hair, photorealistic, high quality"
negative_age = 45
for age in range(10, 91, 5):
print(f"Generating age {age}...")
image = template(
pipe,
prompt=prompt,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": age}],
negative_template_inputs=[{"age": negative_age}],
template_inputs=[{"age": 20}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_{age}.jpg")
image.save(f"image_age_20.jpg")
image = template(
pipe,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": 50}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_50.jpg")
image = template(
pipe,
prompt="A portrait of a woman with black hair, wearing a suit.",
seed=0, cfg_scale=4, num_inference_steps=50,
template_inputs=[{"age": 80}],
negative_template_inputs=[{"age": 45}],
)
image.save(f"image_age_80.jpg")