Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def invoke(prompt):
|
|
31 |
completion = ""
|
32 |
|
33 |
try:
|
34 |
-
|
35 |
model_kwargs = {
|
36 |
"max_tokens": 2048, "temperature": 0.1,
|
37 |
"top_k": 250, "top_p": 1, "stop_sequences": ["\n\nHuman"],
|
|
|
31 |
completion = ""
|
32 |
|
33 |
try:
|
34 |
+
model_id = config["model"]
|
35 |
model_kwargs = {
|
36 |
"max_tokens": 2048, "temperature": 0.1,
|
37 |
"top_k": 250, "top_p": 1, "stop_sequences": ["\n\nHuman"],
|