Tonic commited on
Commit
c4fc3fe
โ€ข
1 Parent(s): 302dd89

add Nemo-Mistral-Minitron / Gradio 5

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ model_info_md = "### Model Configuration: Mistral-NeMo-Minitron-8B-Instruct\n\n"
16
  for key, value in config_info.to_dict().items():
17
  model_info_md += f"- **{key.replace('_', ' ').capitalize()}**: {value}\n"
18
 
19
- pipe = pipeline("text-generation", model=model)
20
- pipe.tokenizer = tokenizer
21
 
22
  def create_prompt(system_message, user_message, tool_definition="", context=""):
23
  if tool_definition:
 
16
  for key, value in config_info.to_dict().items():
17
  model_info_md += f"- **{key.replace('_', ' ').capitalize()}**: {value}\n"
18
 
19
+ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
20
+ # pipe.tokenizer = tokenizer
21
 
22
  def create_prompt(system_message, user_message, tool_definition="", context=""):
23
  if tool_definition: