yrshi commited on
Commit
7d6b435
·
1 Parent(s): 35fef43

update scrpits

Browse files
Files changed (2) hide show
  1. install_env.sh +1 -0
  2. setup.sh +2 -2
install_env.sh CHANGED
@@ -2,6 +2,7 @@
2
  conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main
3
  conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
4
 
 
5
  conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.1 -c pytorch -c nvidia -y
6
  pip install transformers datasets pyserini
7
 
 
2
  conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main
3
  conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
4
 
5
+ conda install python=3.10 -y
6
  conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.1 -c pytorch -c nvidia -y
7
  pip install transformers datasets pyserini
8
 
setup.sh CHANGED
@@ -10,6 +10,6 @@ python download_model.py
10
 
11
  bash retrieval_launch.sh
12
  # sanity check
13
- python infer.py
14
- GRADIO_SERVER_PORT=7890 python app.py
15
  ```
 
10
 
11
  bash retrieval_launch.sh
12
  # sanity check
13
+ # python infer.py
14
+ # GRADIO_SERVER_PORT=7890 python app.py
15
  ```