anonymoussubmitter222 commited on
Commit
b7e34c4
1 Parent(s): 25c18b5

load in the cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -711,6 +711,7 @@ mixer = Mixer(
711
  checkpointer=hparams["checkpointer"],
712
  )
713
  mixer.tokenizer = label_encoder
 
714
  mixer.checkpointer.recover_if_possible()
715
  mixer.modules.eval()
716
 
 
711
  checkpointer=hparams["checkpointer"],
712
  )
713
  mixer.tokenizer = label_encoder
714
+ mixer.device = "cpu"
715
  mixer.checkpointer.recover_if_possible()
716
  mixer.modules.eval()
717