richardblythman
commited on
Commit
•
a51fed4
1
Parent(s):
f458061
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,
|