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

Update app.py (#2)

Browse files

- Update app.py (a20cfd6a07ac865b5bd6197f7e058febd381ea22)


Co-authored-by: huxiaomeng <gregH@users.noreply.huggingface.co>

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"