Fix inference_total
Browse files
app.py
CHANGED
@@ -205,7 +205,7 @@ def calculate_memory(model_name:str, library:str, dtypes:list, optimizer:str, ac
|
|
205 |
vram_activation = vram_f32['activation']
|
206 |
row = {
|
207 |
"dtype": dtype,
|
208 |
-
'inference_total': convert_bytes(vram_model
|
209 |
'training_total': convert_bytes(vram_model + vram_buffer + vram_optimizer + vram_activation),
|
210 |
'model': convert_bytes(vram_model),
|
211 |
'buffer': convert_bytes(vram_buffer),
|
|
|
205 |
vram_activation = vram_f32['activation']
|
206 |
row = {
|
207 |
"dtype": dtype,
|
208 |
+
'inference_total': convert_bytes(vram_model),
|
209 |
'training_total': convert_bytes(vram_model + vram_buffer + vram_optimizer + vram_activation),
|
210 |
'model': convert_bytes(vram_model),
|
211 |
'buffer': convert_bytes(vram_buffer),
|