Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-05-30 19:54:52 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix retry

Browse Source
This commit is contained in:
mertalev 2025-03-14 14:29:07 -04:00
parent 323bcde733
commit 0f1a551842
No known key found for this signature in database
GPG Key ID: F7C271C07CF04AAE
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/export/immich_model_exporter/export.py
View File

@ -4,7 +4,7 @@ import typer
from exporters.constants import DELETE_PATTERNS, SOURCE_TO_METADATA, ModelSource from exporters.constants import DELETE_PATTERNS, SOURCE_TO_METADATA, ModelSource
from exporters.onnx import export as onnx_export from exporters.onnx import export as onnx_export
from exporters.rknn import export as rknn_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 from typing_extensions import Annotated
app = typer.Typer(pretty_exceptions_show_locals=False) app = typer.Typer(pretty_exceptions_show_locals=False)
@ -78,7 +78,7 @@ def main(
repo_id = f"{hf_organization}/{hf_model_name}" 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(): def upload_model():
create_repo(repo_id, exist_ok=True, token=hf_auth_token) create_repo(repo_id, exist_ok=True, token=hf_auth_token)
upload_folder( upload_folder(