Spaces:
Runtime error
Runtime error
Commit ·
169d0fa
1
Parent(s): 2e1968d
fixed cuda bug
Browse files
app.py
CHANGED
|
@@ -18,7 +18,9 @@ DEBIASING_KEYWORDS = [
|
|
| 18 |
|
| 19 |
if torch.cuda.is_available():
|
| 20 |
use_cuda = True
|
| 21 |
-
|
|
|
|
|
|
|
| 22 |
def debias(prompt, model,use_prefix, max_length=50, num_beam=3):
|
| 23 |
"""
|
| 24 |
Debiasing inference function.
|
|
|
|
| 18 |
|
| 19 |
if torch.cuda.is_available():
|
| 20 |
use_cuda = True
|
| 21 |
+
else:
|
| 22 |
+
use_cuda = False
|
| 23 |
+
|
| 24 |
def debias(prompt, model,use_prefix, max_length=50, num_beam=3):
|
| 25 |
"""
|
| 26 |
Debiasing inference function.
|