Upload folder using huggingface_hub
Browse files- chat/run.py +1 -1
chat/run.py
CHANGED
|
@@ -10,7 +10,7 @@ def run(inputs, worker_nodes = None, orchestrator_node = None, flow_run = None,
|
|
| 10 |
{"role": "user", "content": prompt},
|
| 11 |
]
|
| 12 |
response = completion(
|
| 13 |
-
model="ollama/
|
| 14 |
messages=messages,
|
| 15 |
temperature=0,
|
| 16 |
max_tokens=1000,
|
|
|
|
| 10 |
{"role": "user", "content": prompt},
|
| 11 |
]
|
| 12 |
response = completion(
|
| 13 |
+
model="ollama/llama3.1:70b",
|
| 14 |
messages=messages,
|
| 15 |
temperature=0,
|
| 16 |
max_tokens=1000,
|