Spaces:
Runtime error
Runtime error
Commit
·
f36cf47
1
Parent(s):
51b3a2c
Crash?
Browse files
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:")
|