Reza2kn commited on
Commit
6ca1775
·
verified ·
1 Parent(s): dba98b3

Avoid illegal ZeroGPU duration defaults

Browse files
Files changed (1) hide show
  1. app.py +14 -2
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
- DEFAULT_GPU_DURATION = int(_GPU_DURATION)
 
 
 
71
  else:
72
- DEFAULT_GPU_DURATION = 3600 if os.getenv("SPACE_ID") else 0
 
 
 
 
 
 
 
 
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):