Hugging Face's logo

TMElyralab
/
lyraBaichuan

English
LLM
baichuan
Model card Files Community
1
carsonhxsu commited on
Commit
b318031
·
1 Parent(s): fa2f67e

fix model path

Browse files
Files changed (1) hide show
  1. demo.py +2 -2
demo.py CHANGED
@@ -1,7 +1,7 @@
1
  from lyra_baichuan import lyraBaichuan7B, lyraBaichuan13B
2
 
3
- model_path = "./models/Baichuan2-13B-lyra"
4
- tokenizer_path = "./models/Baichuan2-13B-lyra"
5
  inference_dtype = 'fp16'
6
  prompt = "登鹳雀楼->王之涣\n夜雨寄北->"
7
 
 
1
  from lyra_baichuan import lyraBaichuan7B, lyraBaichuan13B
2
 
3
+ model_path = "./models/Baichuan2-13B-Base-lyra"
4
+ tokenizer_path = "./models/Baichuan2-13B-Base-lyra"
5
  inference_dtype = 'fp16'
6
  prompt = "登鹳雀楼->王之涣\n夜雨寄北->"
7