theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:48:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Artiprocher
2026-01-08 13:21:33 +08:00
parent dd479e5bff
commit 8ba528a8f6
4 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
examples/dev_tools/unit_test.py
View File

@@ -108,7 +108,14 @@ def test_flux():
run_inference("examples/flux/model_training/validate_lora")
def test_z_image():
run_inference("examples/z_image/model_inference")
run_inference("examples/z_image/model_inference_low_vram")
run_train_multi_GPU("examples/z_image/model_training/full")
run_inference("examples/z_image/model_training/validate_full")
run_train_single_GPU("examples/z_image/model_training/lora")
run_inference("examples/z_image/model_training/validate_lora")
if __name__ == "__main__":
test_qwen_image()
test_flux()
test_wan()
test_z_image()

6
examples/z_image/model_inference/Z-Image-Omni-Base-i2L.py
View File

@@ -37,11 +37,11 @@ pipe = ZImagePipeline.from_pretrained(
# Load images
snapshot_download(
model_id="DiffSynth-Studio/Qwen-Image-i2L",
model_id="DiffSynth-Studio/Z-Image-Omni-Base-i2L",
allow_file_pattern="assets/style/*",
local_dir="data/examples"
local_dir="data/style_input"
)
images = [Image.open(f"data/style/1/{i}.jpg") for i in range(5)]
images = [Image.open(f"data/style_input/assets/style/1/{i}.jpg") for i in range(6)]
# Image to LoRA
with torch.no_grad():

6
examples/z_image/model_inference_low_vram/Z-Image-Omni-Base-i2L.py
View File

@@ -37,11 +37,11 @@ pipe = ZImagePipeline.from_pretrained(
# Load images
snapshot_download(
model_id="DiffSynth-Studio/Qwen-Image-i2L",
model_id="DiffSynth-Studio/Z-Image-Omni-Base-i2L",
allow_file_pattern="assets/style/*",
local_dir="data/examples"
local_dir="data/style_input"
)
images = [Image.open(f"data/style/1/{i}.jpg") for i in range(5)]
images = [Image.open(f"data/style_input/assets/style/1/{i}.jpg") for i in range(6)]
# Image to LoRA
with torch.no_grad():