fffiloni commited on
Commit
7170008
·
verified ·
1 Parent(s): f7f6486

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ GPU_TO_VRAM_PARAMS = {
82
  "NVIDIA A100-SXM4-40GB": 11000000000,
83
  "NVIDIA A100-SXM4-80GB": 22000000000,
84
  "NVIDIA L4": 5000000000,
85
- "NVIDIA L40S": 11000000000
86
  }
87
  USED_VRAM_PARAMS = GPU_TO_VRAM_PARAMS[gpu_name]
88
  print("Using", USED_VRAM_PARAMS, "for num_persistent_param_in_dit")
 
82
  "NVIDIA A100-SXM4-40GB": 11000000000,
83
  "NVIDIA A100-SXM4-80GB": 22000000000,
84
  "NVIDIA L4": 5000000000,
85
+ "NVIDIA L40S": 22000000000
86
  }
87
  USED_VRAM_PARAMS = GPU_TO_VRAM_PARAMS[gpu_name]
88
  print("Using", USED_VRAM_PARAMS, "for num_persistent_param_in_dit")