injection10 commited on
Commit
cc5e74c
·
verified ·
1 Parent(s): e7b92b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoProcessor, AutoModelForCausalLM
3
  MODEL_ID = "google/gemma-4-31B-it"
4
 
5
  # Load model
6
- processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True
7
  model = AutoModelForCausalLM.from_pretrained(
8
  MODEL_ID,
9
  dtype="auto",
 
3
  MODEL_ID = "google/gemma-4-31B-it"
4
 
5
  # Load model
6
+ processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
7
  model = AutoModelForCausalLM.from_pretrained(
8
  MODEL_ID,
9
  dtype="auto",