Spaces:
Running
on
Zero
Running
on
Zero
add Nemo-Mistral-Minitron / Gradio 5
Browse files
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:
|