support controlnet depth

This commit is contained in:
mi804
2025-08-13 13:36:26 +08:00
parent 34766863f8
commit 6e6136586c
6 changed files with 40 additions and 2 deletions

View File

@@ -18,8 +18,8 @@ pipe = QwenImagePipeline.from_pretrained(
dataset_snapshot_download(
dataset_id="DiffSynth-Studio/example_image_dataset",
local_dir="./",
allow_file_pattern="data/example_image_dataset/canny/image_1.jpg"
local_dir="./data/example_image_dataset",
allow_file_pattern="canny/image_1.jpg"
)
controlnet_image = Image.open("data/example_image_dataset/canny/image_1.jpg").resize((1328, 1328))