Selinaliu1030 commited on
Commit
a9c3fad
·
1 Parent(s): c65dca7

transformer version update

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from huggingface_hub import login
6
  MODEL_ID = "Selinaliu1030/lora_model"
7
 
8
  # Load tokenizer + model
9
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B")
10
 
11
  model = AutoModelForCausalLM.from_pretrained(
12
  MODEL_ID,
 
6
  MODEL_ID = "Selinaliu1030/lora_model"
7
 
8
  # Load tokenizer + model
9
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
10
 
11
  model = AutoModelForCausalLM.from_pretrained(
12
  MODEL_ID,
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
  huggingface_hub
2
  gradio
3
- transformers==4.41.2
 
1
  huggingface_hub
2
  gradio
3
+ transformers>=4.45.0