gregH commited on
Commit
a20cfd6
1 Parent(s): 3e2a079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ m = AutoModelForCausalLM.from_pretrained(
64
  embedding_func=m.get_input_embeddings()
65
  embedding_func.weight.requires_grad=False
66
 
67
- tok = AutoTokenizer.from_pretrained("ibm-granite/granite-3.0-2b-instruct",
68
  trust_remote_code=True,token=HF_TOKEN
69
  )
70
  tok.padding_side = "left"
 
64
  embedding_func=m.get_input_embeddings()
65
  embedding_func.weight.requires_grad=False
66
 
67
+ tok = AutoTokenizer.from_pretrained("google/gemma-2b-it",
68
  trust_remote_code=True,token=HF_TOKEN
69
  )
70
  tok.padding_side = "left"