tb2pi-persistent commited on
Commit
714e176
1 Parent(s): 0fee15b

Added line "tokenizer.pad_token = tokenizer.eos_token"

Browse files
Files changed (1) hide show
  1. 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,