Commit b7bf1144 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 8394d5b3
......@@ -20,6 +20,7 @@ docker exec -it sd-test bash
source /opt/dtk/env.sh
pip3 install click timm
pip3 install diffusers==0.11.1
```
## 推理
#### 01_resnet-50
......@@ -41,7 +42,7 @@ python3 infer_with_torch.py
```
cd examples/02_bert
```
下载bert-base-uncased weights(https://huggingface.co/google-bert/bert-base-uncased)
下载bert-base-uncased weights(http://113.200.138.88:18080/aimodels/bert-base-uncased)
**benchmark**
```
......@@ -56,7 +57,7 @@ python3 demo.py
```
cd examples/03_vit
```
下载vit_base_patch16_224 weights(https://huggingface.co/timm/vit_base_patch16_224.augreg2_in21k_ft_in1k)
下载vit_base_patch16_224 weights(http://113.200.138.88:18080/aimodels/vit_base_patch16_224.augreg2_in21k_ft_in1k)
**benchmark**
```
......@@ -69,9 +70,9 @@ python3 verification.py
```
#### 04_stable_diffusion
下载stable-diffusion-2-1-base weights(https://huggingface.co/stabilityai/stable-diffusion-2-1-base)
下载stable-diffusion-2-1-base weights(http://113.200.138.88:18080/aimodels/stable-diffusion-2-1-base)
下载clip-vit-large-patch14 weights(https://huggingface.co/openai/clip-vit-large-patch14)
下载clip-vit-large-patch14 weights(http://113.200.138.88:18080/aimodels/clip-vit-large-patch14)
**compile**
```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment