Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-05-24 01:12:58 -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.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(