Update app.py
Browse files
app.py
CHANGED
|
@@ -69,6 +69,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 69 |
# device_map=config.DEVICE
|
| 70 |
# )
|
| 71 |
|
|
|
|
| 72 |
def generate(prompt):
|
| 73 |
"""
|
| 74 |
Generate a text completion from a causal language model given a prompt.
|
|
@@ -975,7 +976,6 @@ class BasicAgent:
|
|
| 975 |
if question != "aalskdalsdh" and filename == "":
|
| 976 |
time.sleep(120)
|
| 977 |
|
| 978 |
-
|
| 979 |
state = {
|
| 980 |
"messages": question,
|
| 981 |
}
|
|
|
|
| 69 |
# device_map=config.DEVICE
|
| 70 |
# )
|
| 71 |
|
| 72 |
+
|
| 73 |
def generate(prompt):
|
| 74 |
"""
|
| 75 |
Generate a text completion from a causal language model given a prompt.
|
|
|
|
| 976 |
if question != "aalskdalsdh" and filename == "":
|
| 977 |
time.sleep(120)
|
| 978 |
|
|
|
|
| 979 |
state = {
|
| 980 |
"messages": question,
|
| 981 |
}
|