def push_to_hub()

in distilvit/upload.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

def push_to_hub(model_id, save_path, tag, commit_message):
    api = HfApi()

    api.upload_folder(
        repo_id=model_id, folder_path=save_path, commit_message=commit_message
    )

    if args.tag:
        api.create_tag(
            repo_id=model_id,
            tag=tag,
        )