Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,8 +15,8 @@ path_to_log = "FlaggedFalse.txt"
|
|
| 15 |
|
| 16 |
# Inference client with the model (And HF-token if needed)
|
| 17 |
client = InferenceClient(checkpoint)
|
| 18 |
-
tokenizer = AutoTokenizer.from_pretrained(checkpoint
|
| 19 |
-
if checkpoint == "NousResearch/Nous-
|
| 20 |
# Tokenizer chat template correction(Only works for mistral models)
|
| 21 |
chat_template = open("mistral-instruct.jinja").read()
|
| 22 |
chat_template = chat_template.replace(' ', '').replace('\n', '')
|
|
|
|
| 15 |
|
| 16 |
# Inference client with the model (And HF-token if needed)
|
| 17 |
client = InferenceClient(checkpoint)
|
| 18 |
+
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 19 |
+
if checkpoint == "NousResearch/Nous-He-8x7B-DPO":
|
| 20 |
# Tokenizer chat template correction(Only works for mistral models)
|
| 21 |
chat_template = open("mistral-instruct.jinja").read()
|
| 22 |
chat_template = chat_template.replace(' ', '').replace('\n', '')
|