Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ def load_models():
|
|
23 |
tokenizer_path=f'{weight_path}/tokenizer.model',
|
24 |
max_seq_len=100,
|
25 |
max_batch_size=32,
|
|
|
26 |
model_parallel_size=1)
|
27 |
return model
|
28 |
|
|
|
23 |
tokenizer_path=f'{weight_path}/tokenizer.model',
|
24 |
max_seq_len=100,
|
25 |
max_batch_size=32,
|
26 |
+
device="cpu",
|
27 |
model_parallel_size=1)
|
28 |
return model
|
29 |
|