Update README.md

README.md

@@ -6,6 +6,7 @@ language:
 - en
 ---
 
+
 ## Using HunyuanDiT ControlNet
 
 
@@ -20,7 +21,9 @@ language:
 ```bash
 cd HunyuanDiT
 # Use the huggingface-cli tool to download the model.
+# We recommend using distilled weights as the base model for ControlNet inference, as our provided pretrained weights are trained on them.
 huggingface-cli download Tencent-Hunyuan/HYDiT-ControlNet --local-dir ./ckpts/t2i/controlnet
+huggingface-cli download Tencent-Hunyuan/Distillation-v1.1 ./pytorch_model_distill.pt --local-dir ./ckpts/t2i/model
 
 # Quick start
 python3 sample_controlnet.py  --no-enhance --load-key distill --infer-steps 50 --control_type canny --prompt "在夜晚的酒店门前，一座古老的中国风格的狮子雕像矗立着，它的眼睛闪烁着光芒，仿佛在守护着这座建筑。背景是夜晚的酒店前，构图方式是特写，平视，居中构图。这张照片呈现了真实摄影风格，蕴含了中国雕塑文化，同时展现了神秘氛围" --condition_image_path controlnet/asset/input/canny.jpg --control_weight 1.0
@@ -67,10 +70,17 @@ Examples of condition input and ControlNet results are as follows:
 
 ### Training
 
-We utilize [**DWPose**](https://github.com/IDEA-Research/DWPose) for pose extraction. Please follow their guidelines to download the checkpoints and save them to `hydit/annotator/ckpts` directory. Additionally, ensure that you install the related dependencies.
+We utilize [**DWPose**](https://github.com/IDEA-Research/DWPose) for pose extraction. Please follow their guidelines to download the checkpoints and save them to `hydit/annotator/ckpts` directory. We provide serveral commands to quick install:
+```bash
+mkdir ./hydit/annotator/ckpts
+wget -O ./hydit/annotator/ckpts/dwpose.zip https://dit.hunyuan.tencent.com/download/HunyuanDiT/dwpose.zip
+unzip ./hydit/annotator/ckpts/dwpose.zip -d ./hydit/annotator/ckpts/
+```
+Additionally, ensure that you install the related dependencies.
 ```bash
-pip install matplotlib
-pip install onnxruntime_gpu
+pip install matplotlib==3.7.5
+pip install onnxruntime_gpu==1.16.3
+pip install opencv-python==4.8.1.78
 ```
 
 
@@ -162,3 +172,4 @@ c. Using depth ControlNet during inference
 python3 sample_controlnet.py  --no-enhance --load-key distill --infer-steps 50 --control_type pose --prompt "一位亚洲女性，身穿绿色上衣，戴着紫色头巾和紫色围巾，站在黑板前。背景是黑板。照片采用近景、平视和居中构图的方式呈现真实摄影风格" --condition_image_path controlnet/asset/input/pose.jpg --control_weight 1.0
 ```
 
+