Spaces:

selbl
/

LyricGeneration

Sleeping

selbl commited on Feb 24

Commit

4a37fb4

•

1 Parent(s): 637f1ec

Update StreamlitModel.py

Files changed (1) hide show

StreamlitModel.py CHANGED Viewed

@@ -19,7 +19,9 @@ import torch.hub
 profanity.load_censor_words()
-device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_built() else 'cpu'
 class GPT2_Model(GPT2PreTrainedModel):

 profanity.load_censor_words()
+#It seems the streamlit space does not allow mps
+#device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_built() else 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 class GPT2_Model(GPT2PreTrainedModel):