Update app.py
Browse files
app.py
CHANGED
|
@@ -227,7 +227,7 @@ def _warmup_thread(models: List[str]):
|
|
| 227 |
t_global = time.time()
|
| 228 |
_log_warmup(f"[JOB] start • {len(models)} dépôts")
|
| 229 |
|
| 230 |
-
for i, repo in enumerate(models):
|
| 231 |
if warmup_stop.is_set():
|
| 232 |
_log_warmup("[STOP] Arrêt demandé — fin du job après ce point")
|
| 233 |
break
|
|
@@ -247,7 +247,7 @@ for i, repo in enumerate(models):
|
|
| 247 |
elif res == "cache":
|
| 248 |
warmup_state.setdefault("cache_repos", []).append(repo)
|
| 249 |
elif res == "fail":
|
| 250 |
-
warmup_state
|
| 251 |
elif res == "stopped":
|
| 252 |
_log_warmup("[STOP] Fin anticipée — demande reçue pendant le run")
|
| 253 |
warmup_state["percent"] = int(((i + 1) / max(1, len(models))) * 100)
|
|
@@ -259,14 +259,15 @@ for i, repo in enumerate(models):
|
|
| 259 |
_log_warmup("[STOP] Fin anticipée — demande reçue pendant le run")
|
| 260 |
break
|
| 261 |
|
| 262 |
-
|
| 263 |
-
|
| 264 |
warmup_state["done"] = True
|
| 265 |
warmup_state["running"] = False
|
| 266 |
warmup_state["ok_count"] = ok_count
|
| 267 |
warmup_state["finished_at"] = time.time()
|
| 268 |
|
| 269 |
-
|
|
|
|
| 270 |
# <<<< HUGINFACE PATCH: WARMUP STATE+HELPERS END >>>>
|
| 271 |
# ---------- Helpers ----------
|
| 272 |
|
|
|
|
| 227 |
t_global = time.time()
|
| 228 |
_log_warmup(f"[JOB] start • {len(models)} dépôts")
|
| 229 |
|
| 230 |
+
for i, repo in enumerate(models):
|
| 231 |
if warmup_stop.is_set():
|
| 232 |
_log_warmup("[STOP] Arrêt demandé — fin du job après ce point")
|
| 233 |
break
|
|
|
|
| 247 |
elif res == "cache":
|
| 248 |
warmup_state.setdefault("cache_repos", []).append(repo)
|
| 249 |
elif res == "fail":
|
| 250 |
+
warmup_state.setdefault("failed_repos", []).append(repo)
|
| 251 |
elif res == "stopped":
|
| 252 |
_log_warmup("[STOP] Fin anticipée — demande reçue pendant le run")
|
| 253 |
warmup_state["percent"] = int(((i + 1) / max(1, len(models))) * 100)
|
|
|
|
| 259 |
_log_warmup("[STOP] Fin anticipée — demande reçue pendant le run")
|
| 260 |
break
|
| 261 |
|
| 262 |
+
with warmup_lock:
|
| 263 |
+
warmup_state["percent"] = 100
|
| 264 |
warmup_state["done"] = True
|
| 265 |
warmup_state["running"] = False
|
| 266 |
warmup_state["ok_count"] = ok_count
|
| 267 |
warmup_state["finished_at"] = time.time()
|
| 268 |
|
| 269 |
+
_log_warmup(f"[JOB] done • {ok_count}/{len(models)} • {time.time()-t_global:.1f}s")
|
| 270 |
+
|
| 271 |
# <<<< HUGINFACE PATCH: WARMUP STATE+HELPERS END >>>>
|
| 272 |
# ---------- Helpers ----------
|
| 273 |
|