Update app.py
Browse files
app.py
CHANGED
|
@@ -60,8 +60,7 @@ class EventScraper:
|
|
| 60 |
model_name,
|
| 61 |
torch_dtype=torch.float16,
|
| 62 |
return_dict_in_generate=False,
|
| 63 |
-
device_map='auto'
|
| 64 |
-
return_full_text=False
|
| 65 |
)
|
| 66 |
return
|
| 67 |
except Exception as local_err:
|
|
|
|
| 60 |
model_name,
|
| 61 |
torch_dtype=torch.float16,
|
| 62 |
return_dict_in_generate=False,
|
| 63 |
+
device_map='auto'
|
|
|
|
| 64 |
)
|
| 65 |
return
|
| 66 |
except Exception as local_err:
|