FABLESLIP commited on
Commit
d5d1216
·
verified ·
1 Parent(s): 030d017

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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["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,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
- 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
  # <<<< 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