utensil commited on
Commit
8a43e43
1 Parent(s): f3c24fd

Fix inference_total

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 + vram_activation),
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),