2024-02-11 08:06:50 +00:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
|
set -exu
|
|
|
|
|
|
2024-02-12 11:40:51 +00:00
|
|
|
export CHECKPOINT_DIR="default"
|
|
|
|
|
export TOKENIZER_DIR="default"
|
2024-03-25 06:54:22 +00:00
|
|
|
# export USE_CUDA=False # True or False (case-sensitive)
|
|
|
|
|
export USE_CUDA=True # True or False (case-sensitive)
|
|
|
|
|
export NUM_BEAM=10
|
2024-02-11 08:06:50 +00:00
|
|
|
|
|
|
|
|
streamlit run web.py
|
