Ctaake commited on
Commit
1612461
·
verified ·
1 Parent(s): 6007798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,force_download=True)
19
- if checkpoint == "NousResearch/Nous-Hermes-2-Mixtral-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', '')
 
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', '')