Ctaake commited on
Commit
6d0a1ba
·
verified ·
1 Parent(s): f3d9b35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from mySystemPrompt import SYSTEM_PROMPT
8
  checkpoint = "CohereForAI/c4ai-command-r-plus"
9
  # Inference client with the model (And HF-token if needed)
10
  client = InferenceClient(checkpoint)
11
- tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
  # Tokenizer chat template correction(Only works for mistral models)
13
  #chat_template = open("mistral-instruct.jinja").read()
14
  #chat_template = chat_template.replace(' ', '').replace('\n', '')
 
8
  checkpoint = "CohereForAI/c4ai-command-r-plus"
9
  # Inference client with the model (And HF-token if needed)
10
  client = InferenceClient(checkpoint)
11
+ tokenizer = AutoTokenizer.from_pretrained("prince-canuma/c4ai-command-r-v01-tokenizer-chat-template")
12
  # Tokenizer chat template correction(Only works for mistral models)
13
  #chat_template = open("mistral-instruct.jinja").read()
14
  #chat_template = chat_template.replace(' ', '').replace('\n', '')