Spaces:
Sleeping
Sleeping
Commit
•
5824320
1
Parent(s):
79a5328
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def load_models():
|
|
22 |
tokenizer_path=f'{weight_path}/tokenizer.model',
|
23 |
max_seq_len=100,
|
24 |
max_batch_size=32,
|
25 |
-
device="
|
26 |
model_parallel_size=1)
|
27 |
return model
|
28 |
|
|
|
22 |
tokenizer_path=f'{weight_path}/tokenizer.model',
|
23 |
max_seq_len=100,
|
24 |
max_batch_size=32,
|
25 |
+
device="cuda",
|
26 |
model_parallel_size=1)
|
27 |
return model
|
28 |
|