Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
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":
|
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")
|