Locutusque commited on
Commit
d168895
·
verified ·
1 Parent(s): c77e10a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ def generate(
21
  pipe = load_model(model_name)
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
23
  print(tokenizer)
 
24
 
25
  # Set tokenize correctly. Otherwise ticking the box breaks it.
26
  if model_name == "M4-ai/tau-1.8B":
 
21
  pipe = load_model(model_name)
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
23
  print(tokenizer)
24
+ pipe.tokenizer = tokenizer
25
 
26
  # Set tokenize correctly. Otherwise ticking the box breaks it.
27
  if model_name == "M4-ai/tau-1.8B":