youssef227 commited on
Commit
8b79802
·
verified ·
1 Parent(s): c854b77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,9 +3,9 @@ from peft import LoraConfig, get_peft_model
3
  import torch
4
 
5
  # Load the tokenizer and model
6
- tokenizer = AutoTokenizer.from_pretrained("youssef227/llama-3-8b-Instruct-bnb-telcom-3")
7
  print("step 1 ")
8
- model = AutoModelForCausalLM.from_pretrained("youssef227/llama-3-8b-Instruct-bnb-telcom-3")
9
  print("step 2 ")
10
 
11
  def generator(text):
 
3
  import torch
4
 
5
  # Load the tokenizer and model
6
+ model = AutoModelForCausalLM.from_pretrained("youssef227/llama-3-8b-Instruct-bnb-telcom-3",hf=True)
7
  print("step 1 ")
8
+ tokenizer = AutoTokenizer.from_pretrained(model)
9
  print("step 2 ")
10
 
11
  def generator(text):