ethanlshen commited on
Commit
2b84629
·
verified ·
1 Parent(s): 3419da4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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