File size: 389 Bytes
a763090
70a9c15
a763090
 
 
 
 
 
70a9c15
90a7cda
67bfb24
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
import os

os.system("pwd")
#os.system("nvidia-smi")
os.system(
    f"git clone https://github.com/icalk-nlp/EduChat.git")
#os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
#os.system(f"cd EduChat")

os.system(f'''python launch.py  --model_path ecnu-icalk/educhat-sft-002-7b \
--top_k 50 \
--do_sample True \
--max_new_tokens 512''')