Spaces:

B-K
/

song2midi

Sleeping

B-K commited on Jun 10, 2024

Commit

17ea062

verified ·

1 Parent(s): f0d52bd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -219,7 +219,7 @@ class ReformerEncoderDecoder(PreTrainedModel):
 # model load
 model = ReformerEncoderDecoder(ReformerEncoderDecoderConfig()).to(device)
-model.load_state_dict(torch.load("model.pth"), map_location=torch.device(device))
 tokenizer = REMI(params="tokenizer.json")
 # model load end

 # model load
 model = ReformerEncoderDecoder(ReformerEncoderDecoderConfig()).to(device)
+model.load_state_dict(torch.load("model.pth", map_location=torch.device(device)))
 tokenizer = REMI(params="tokenizer.json")
 # model load end