ejschwartz commited on
Commit
f36cf47
·
1 Parent(s): 51b3a2c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ sample_param = next(vardecoder_model.parameters())
55
  print(f"Sample parameter shape: {sample_param.shape}")
56
  print(f"Sample parameter requires_grad: {sample_param.requires_grad}")
57
  print(f"Sample parameter data type: {type(sample_param.data)}")
58
- print(f"Sample parameter storage: {sample_param.storage()}")
59
 
60
  # Check memory after first model
61
  print(f"GPU memory after vardecoder:")
 
55
  print(f"Sample parameter shape: {sample_param.shape}")
56
  print(f"Sample parameter requires_grad: {sample_param.requires_grad}")
57
  print(f"Sample parameter data type: {type(sample_param.data)}")
58
+ #print(f"Sample parameter storage: {sample_param.storage()}")
59
 
60
  # Check memory after first model
61
  print(f"GPU memory after vardecoder:")