bstraehle commited on
Commit
331094f
1 Parent(s): c06669a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -138,7 +138,7 @@ def prompt_model(model_name, system_prompt, user_prompt, sql_schema):
138
  def load_model(model_name):
139
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
140
  tokenizer = AutoTokenizer.from_pretrained(model_name)
141
- #tokenizer.pad_token = tokenizer.eos_token
142
 
143
  return model, tokenizer
144
 
 
138
  def load_model(model_name):
139
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
140
  tokenizer = AutoTokenizer.from_pretrained(model_name)
141
+ tokenizer.pad_token = tokenizer.eos_token
142
 
143
  return model, tokenizer
144