Laurie commited on
Commit
6828abc
·
1 Parent(s): 6127e5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -17,4 +17,8 @@
17
  model = model.cuda()
18
  model.eval()
19
 
20
- # model.generate, model.chat()...
 
 
 
 
 
17
  model = model.cuda()
18
  model.eval()
19
 
20
+ # model.generate, model.chat()...
21
+
22
+ # 命令行进行测试
23
+ CUDA_VISIBLE_DEVICES=0 python src/infer.py \
24
+ --checkpoint_dir path_to_checkpoint