Hugging Face's logo

Spaces:
fffiloni
/
Meigen-MultiTalk
Running on L40S

App Files Community
fffiloni commited on
Commit
98659d3
·
verified ·
1 Parent(s): da0d3cc

L40S param

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -91,7 +91,8 @@ GPU_TO_VRAM_PARAMS = {
91
  "NVIDIA A100": 11000000000,
92
  "NVIDIA A100-SXM4-40GB": 11000000000,
93
  "NVIDIA A100-SXM4-80GB": 22000000000,
94
- "NVIDIA L4": 5000000000
 
95
  }
96
  USED_VRAM_PARAMS = GPU_TO_VRAM_PARAMS[gpu_name]
97
  print("Using", USED_VRAM_PARAMS, "for num_persistent_param_in_dit")
 
91
  "NVIDIA A100": 11000000000,
92
  "NVIDIA A100-SXM4-40GB": 11000000000,
93
  "NVIDIA A100-SXM4-80GB": 22000000000,
94
+ "NVIDIA L4": 5000000000,
95
+ "NVIDIA L40S": 5000000000
96
  }
97
  USED_VRAM_PARAMS = GPU_TO_VRAM_PARAMS[gpu_name]
98
  print("Using", USED_VRAM_PARAMS, "for num_persistent_param_in_dit")