Update README.md

README.md

@@ -14,17 +14,14 @@ stable diffusion的核心是latent diffusion model,latent diffusion model结构
 ## 环境配置
 在[光源](https://sourcefind.cn/#/service-list)可拉取推理的docker镜像。stable_diffusion_ait推荐的镜像如下：
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ait-0.0.1_dtk24.04_py310
-docker run -d -t -v /opt/hyhal:/opt/hyhal:ro --privileged --device=/dev/kfd --device=/dev/dri/ --network=host --group-add video --name sd-test image.sourcefind.cn:5000/dcu/admin/base/custom:ait-0.0.1_dtk24.04_py310
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
+docker run -d -t -v /opt/hyhal:/opt/hyhal:ro --privileged --device=/dev/kfd --device=/dev/dri/ --network=host --group-add video --name sd-test image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 docker exec -it sd-test bash
 source /opt/dtk/env.sh
+
+pip3 install click timm 
 ```
 ## 推理
-**install ait**
-```
-cd stable_diffusion_ait
-pip3 install dist/aitemplate-0.0.1-py3-none-any.whl
-```
 #### 01_resnet-50
 ```
 cd examples/01_resnet-50