From 0f1a551842aba7934b1e629655b80fa05f6a92e6 Mon Sep 17 00:00:00 2001
From: mertalev <101130780+mertalev@users.noreply.github.com>
Date: Fri, 14 Mar 2025 14:29:07 -0400
Subject: [PATCH] fix retry

---
 machine-learning/export/immich_model_exporter/export.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/machine-learning/export/immich_model_exporter/export.py b/machine-learning/export/immich_model_exporter/export.py
index 2b4fb52b18..0d2ae735ba 100644
--- a/machine-learning/export/immich_model_exporter/export.py
+++ b/machine-learning/export/immich_model_exporter/export.py
@@ -4,7 +4,7 @@ import typer
 from exporters.constants import DELETE_PATTERNS, SOURCE_TO_METADATA, ModelSource
 from exporters.onnx import export as onnx_export
 from exporters.rknn import export as rknn_export
-from tenacity import retry, stop_after_attempt
+from tenacity import retry, stop_after_attempt, wait_fixed
 from typing_extensions import Annotated
 
 app = typer.Typer(pretty_exceptions_show_locals=False)
@@ -78,7 +78,7 @@ def main(
 
         repo_id = f"{hf_organization}/{hf_model_name}"
 
-        @retry(stop=stop_after_attempt(3), wait=2)
+        @retry(stop=stop_after_attempt(5), wait=wait_fixed(5))
         def upload_model():
             create_repo(repo_id, exist_ok=True, token=hf_auth_token)
             upload_folder(