jpterry commited on
Commit
d1afb30
·
1 Parent(s): e5c4282

printing a whole bunch of bullshit

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -238,7 +238,11 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
238
  depth_mult=hparams.depth_mult,)
239
 
240
  model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
241
- model.load_state_dict(torch.load(model_url, map_location='cpu'))
 
 
 
 
242
 
243
  print(model)
244
 
 
238
  depth_mult=hparams.depth_mult,)
239
 
240
  model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
241
+
242
+ loaded = torch.load(model_url, map_location='cpu')
243
+ print(loaded)
244
+
245
+ model.load_state_dict(loaded)
246
 
247
  print(model)
248