Hugging Face's logo

Spaces:
JasonTPhillipsJr
/
SpaGAN
Sleeping

App Files Community
JasonTPhillipsJr commited on
Commit
811f100
·
verified ·
1 Parent(s): 0a51000

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ config.output_hidden_states = True
30
  spaBERT_model = SpatialBertForMaskedLM(config)
31
  spaBERT_model.load_state_dict(b_model.state_dict(), strict = False)
32
 
33
- pre_trained_model = torch.load(pretrained_model)
34
 
35
 
36
 
 
30
  spaBERT_model = SpatialBertForMaskedLM(config)
31
  spaBERT_model.load_state_dict(b_model.state_dict(), strict = False)
32
 
33
+ pre_trained_model = torch.load(pretrained_model, map_location=torch.device('cpu'))
34
 
35
 
36