Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

import requests
from qiniu import Auth, put_file, put_stream_v2, put_data
from qiniu import config as qiniu_config
from jinja2 import Environment, BaseLoader
from huggingface_hub import HfApi, hf_hub_url
from huggingface_hub.hf_api import RepoFolder
Expand Down Expand Up @@ -142,6 +143,10 @@ def upload_hf_repo(


if __name__ == '__main__':
# Configure longer timeout for large file uploads (5 minutes instead of default 30s)
# This prevents TimeoutError when uploading large files like model.bin through qiniu's put_stream_v2
qiniu_config.set_default(connection_timeout=60*60*3)

parser = argparse.ArgumentParser()
parser.add_argument('--ak', type=str, help='qiniu access key')
parser.add_argument('--sk', type=str, help='qiniu secret key')
Expand Down