Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -42,6 +42,7 @@ start_messages = [
|
|
42 |
{'role': 'assistant', 'content': '<Waiting for text>'}
|
43 |
]
|
44 |
|
|
|
45 |
|
46 |
@spaces.GPU
|
47 |
def get_past_key_values(system_prompt):
|
|
|
42 |
{'role': 'assistant', 'content': '<Waiting for text>'}
|
43 |
]
|
44 |
|
45 |
+
torch.set_grad_enabled(False)
|
46 |
|
47 |
@spaces.GPU
|
48 |
def get_past_key_values(system_prompt):
|