Update README.md

README.md

@@ -8,13 +8,13 @@ license_link: https://huggingface.co/Tencent-Hunyuan/HunyuanDiT/blob/main/LICENS
 from diffusers import HunyuanDiT2DControlNetModel, HunyuanDiTControlNetPipeline
 import torch
 
-controlnet = HunyuanDiT2DControlNetModel.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Distilled-Depth", torch_dtype=torch.float16)
+controlnet = HunyuanDiT2DControlNetModel.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Depth", torch_dtype=torch.float16)
 
 pipe = HunyuanDiTControlNetPipeline.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-Diffusers-Distilled", controlnet=controlnet, torch_dtype=torch.float16)
 pipe.to("cuda")
 
 from diffusers.utils import load_image
-cond_image = load_image('https://huggingface.co/Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Distilled-Depth/resolve/main/depth.jpg?download=true')
+cond_image = load_image('https://huggingface.co/Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Depth/resolve/main/depth.jpg?download=true')
 
 ## You may also use English prompt as HunyuanDiT supports both English and Chinese
 prompt="在茂密的森林中，一只黑白相间的熊猫静静地坐在绿树红花中，周围是山川和海洋。背景是白天的森林，光线充足。照片采用特写、平视和居中构图的方式，呈现出写实的效果"
@@ -30,4 +30,6 @@ image = pipe(
     control_image=cond_image,
     num_inference_steps=50,
 ).images[0]
+
+image.save('./image.png')
 ```