Spaces:
Running
on
L40S
Running
on
L40S
File size: 430 Bytes
258fd02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
export USER=root
export PYTHONDONTWRITEBYTECODE=1
export TRANSFORMERS_CACHE="$(pwd)/third_party/hub"
export NCCL_HOME=/usr/local/tccl
export PYTHONPATH="$(pwd)/codeclm/tokenizer/":"$(pwd)":"$(pwd)/codeclm/tokenizer/Flow1dVAE/":"$(pwd)/codeclm/tokenizer/":$PYTHONPATH
CFG_FILE=conf/infer.yaml
JSONL=$1
SAVE_DIR=$2
SIDX=0
DEVICE=0
OMP_NUM_THREADS=1 CUDA_VISIBLE_DEVICES=$DEVICE python3 generate.py $CFG_FILE $SAVE_DIR $JSONL $SIDX
|