Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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":
|