Spaces:
Running on Zero
Running on Zero
Avoid illegal ZeroGPU duration defaults
Browse files
app.py
CHANGED
|
@@ -66,10 +66,22 @@ def log_progress(message: str, level: int = 1, enabled: bool = True) -> None:
|
|
| 66 |
load_dotenv()
|
| 67 |
|
| 68 |
_GPU_DURATION = os.getenv("CHIZZLER_GPU_DURATION")
|
|
|
|
| 69 |
if _GPU_DURATION is not None:
|
| 70 |
-
|
|
|
|
|
|
|
|
|
|
| 71 |
else:
|
| 72 |
-
DEFAULT_GPU_DURATION =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
|
| 74 |
|
| 75 |
def gpu_decorator(duration: int):
|
|
|
|
| 66 |
load_dotenv()
|
| 67 |
|
| 68 |
_GPU_DURATION = os.getenv("CHIZZLER_GPU_DURATION")
|
| 69 |
+
_GPU_DURATION_MAX = os.getenv("CHIZZLER_GPU_DURATION_MAX")
|
| 70 |
if _GPU_DURATION is not None:
|
| 71 |
+
try:
|
| 72 |
+
DEFAULT_GPU_DURATION = int(_GPU_DURATION)
|
| 73 |
+
except ValueError:
|
| 74 |
+
DEFAULT_GPU_DURATION = 0
|
| 75 |
else:
|
| 76 |
+
DEFAULT_GPU_DURATION = 0
|
| 77 |
+
|
| 78 |
+
if _GPU_DURATION_MAX is not None:
|
| 79 |
+
try:
|
| 80 |
+
max_duration = int(_GPU_DURATION_MAX)
|
| 81 |
+
if max_duration > 0:
|
| 82 |
+
DEFAULT_GPU_DURATION = min(DEFAULT_GPU_DURATION, max_duration)
|
| 83 |
+
except ValueError:
|
| 84 |
+
pass
|
| 85 |
|
| 86 |
|
| 87 |
def gpu_decorator(duration: int):
|