Skip to content

GitLab

Commit b7bf1144 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 8394d5b3
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
README.md
View file @ b7bf1144
...@@ -20,6 +20,7 @@ docker exec -it sd-test bash ...@@ -20,6 +20,7 @@ docker exec -it sd-test bash
source /opt/dtk/env.sh source /opt/dtk/env.sh
pip3 install click timm pip3 install click timm
pip3 install diffusers==0.11.1
``` ```
## 推理 ## 推理
#### 01_resnet-50 #### 01_resnet-50
...@@ -41,7 +42,7 @@ python3 infer_with_torch.py ...@@ -41,7 +42,7 @@ python3 infer_with_torch.py
``` ```
cd examples/02_bert 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** **benchmark**
``` ```
...@@ -56,7 +57,7 @@ python3 demo.py ...@@ -56,7 +57,7 @@ python3 demo.py
``` ```
cd examples/03_vit 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** **benchmark**
``` ```
...@@ -69,9 +70,9 @@ python3 verification.py ...@@ -69,9 +70,9 @@ python3 verification.py
``` ```
#### 04_stable_diffusion #### 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** **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