tianyang commited on
Commit
3cb6909
·
1 Parent(s): 70ebc84
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from utils.gradio import reset_textbox, cancel_outputing, transfer_input, \
12
 
13
  # set variables
14
  BASE_MODEL = "llama-7B"
15
- LORA_MODEL = "output/llama_lora_7B/checkpoint-3700"
16
 
17
 
18
  print("Loading model...")
 
12
 
13
  # set variables
14
  BASE_MODEL = "llama-7B"
15
+ LORA_MODEL = "lemur-7B"
16
 
17
 
18
  print("Loading model...")
requirements.txt CHANGED
@@ -3,4 +3,5 @@ torch==2.0.1
3
  git+https://github.com/huggingface/transformers.git
4
  gradio
5
  datasets
6
- bitsandbytes
 
 
3
  git+https://github.com/huggingface/transformers.git
4
  gradio
5
  datasets
6
+ bitsandbytes
7
+ scipy