seko02 commited on
Commit
a463500
·
verified ·
1 Parent(s): 95263c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
5
  from peft import PeftModel
6
 
7
  BASE_MODEL = "unsloth/Llama-3.2-3B-Instruct"
8
- LORA_REPO = "./3B/" # Change this to 1B to use smaller model
9
  device = "cpu"
10
 
11
  print('loading tokenizer')
 
5
  from peft import PeftModel
6
 
7
  BASE_MODEL = "unsloth/Llama-3.2-3B-Instruct"
8
+ LORA_REPO = "./1B/" # Change this to 1B to use smaller model
9
  device = "cpu"
10
 
11
  print('loading tokenizer')