youssef227 commited on
Commit
c854b77
·
verified ·
1 Parent(s): 2e31dfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -3,9 +3,10 @@ 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", trust_remote_code=True)
7
- model = AutoModelForCausalLM.from_pretrained("youssef227/llama-3-8b-Instruct-bnb-telcom-3", trust_remote_code=True)
8
-
 
9
 
10
  def generator(text):
11
  inputs = tokenizer(
 
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):
12
  inputs = tokenizer(