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

fix validate full

Browse Source
This commit is contained in:
mi804
2025-08-18 16:17:40 +08:00
parent 398b1dbd7a
commit ad1da43476
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/qwen_image/model_training/validate_full/Qwen-Image-Edit.py
View File

@@ -15,6 +15,7 @@ pipe = QwenImagePipeline.from_pretrained(
processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit", origin_file_pattern="processor/"), processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit", origin_file_pattern="processor/"),
) )
state_dict = load_state_dict("models/train/Qwen-Image-Edit_full/epoch-1.safetensors") state_dict = load_state_dict("models/train/Qwen-Image-Edit_full/epoch-1.safetensors")
pipe.dit.load_state_dict(state_dict)
prompt = "将裙子改为粉色" prompt = "将裙子改为粉色"
image = Image.open("data/example_image_dataset/edit/image1.jpg").resize((1024, 1024)) image = Image.open("data/example_image_dataset/edit/image1.jpg").resize((1024, 1024))