Update app.py (#2)
Browse files- Update app.py (a20cfd6a07ac865b5bd6197f7e058febd381ea22)
Co-authored-by: huxiaomeng <gregH@users.noreply.huggingface.co>
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("
|
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"
|