tb2pi-persistent
commited on
Commit
•
714e176
1
Parent(s):
0fee15b
Added line "tokenizer.pad_token = tokenizer.eos_token"
Browse files- handler.py +1 -0
handler.py
CHANGED
@@ -9,6 +9,7 @@ class EndpointHandler:
|
|
9 |
def __init__(self, path=""):
|
10 |
|
11 |
tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True)
|
|
|
12 |
model = AutoModelForCausalLM.from_pretrained(
|
13 |
path,
|
14 |
return_dict=True,
|
|
|
9 |
def __init__(self, path=""):
|
10 |
|
11 |
tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True)
|
12 |
+
tokenizer.pad_token = tokenizer.eos_token
|
13 |
model = AutoModelForCausalLM.from_pretrained(
|
14 |
path,
|
15 |
return_dict=True,
|