Spaces:
Running
on
T4
Running
on
T4
Ahmet Kaan Sever
commited on
Commit
·
db96c4e
1
Parent(s):
2dfc1fc
Only allow t4-medium
Browse files- svc/router.py +1 -1
svc/router.py
CHANGED
|
@@ -137,7 +137,7 @@ def get_gpu_tier():
|
|
| 137 |
# Normalize GPU model to your custom tier system
|
| 138 |
if "t4" in gpu_name:
|
| 139 |
# You can improve this by checking memory or other context
|
| 140 |
-
return {"gpu": "Tesla T4", "tier": "t4-
|
| 141 |
elif "l4" in gpu_name:
|
| 142 |
return {"gpu": "NVIDIA L4", "tier": "l4x1"}
|
| 143 |
elif "l40s" in gpu_name:
|
|
|
|
| 137 |
# Normalize GPU model to your custom tier system
|
| 138 |
if "t4" in gpu_name:
|
| 139 |
# You can improve this by checking memory or other context
|
| 140 |
+
return {"gpu": "Tesla T4", "tier": "t4-medium"}
|
| 141 |
elif "l4" in gpu_name:
|
| 142 |
return {"gpu": "NVIDIA L4", "tier": "l4x1"}
|
| 143 |
elif "l40s" in gpu_name:
|