CallmeKaito commited on
Commit
dfc7b48
·
verified ·
1 Parent(s): bdd39e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from peft import PeftModel
8
  def load_model():
9
  base_model = AutoModelForCausalLM.from_pretrained("unsloth/Llama-3.2-1B-Instruct")
10
  model = PeftModel.from_pretrained(base_model, "CallmeKaito/llama-3.2-1b-it-brainrot")
11
- tokenizer = AutoTokenizer.from_pretrained("unslosh/Llama-3.2-1B-Instruct")
12
  model = model.to("cuda")
13
  return model, tokenizer
14
 
 
8
  def load_model():
9
  base_model = AutoModelForCausalLM.from_pretrained("unsloth/Llama-3.2-1B-Instruct")
10
  model = PeftModel.from_pretrained(base_model, "CallmeKaito/llama-3.2-1b-it-brainrot")
11
+ tokenizer = AutoTokenizer.from_pretrained("unsloth/Llama-3.2-1B-Instruct")
12
  model = model.to("cuda")
13
  return model, tokenizer
14