|
|
|
|
|
import argparse |
|
|
import os |
|
|
|
|
|
from huggingface_hub import HfApi |
|
|
|
|
|
import model_tools as mt |
|
|
|
|
|
REPO_OWNER = "lainlives" |
|
|
REPO_NAME = "voice" |
|
|
repo_type = "model" |
|
|
HF_TOKEN = os.getenv("HF_TOKEN") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
repo_id = f"{REPO_OWNER}/{REPO_NAME}" |
|
|
|
|
|
parser = argparse.ArgumentParser(description="Model Upload Tool") |
|
|
parser.add_argument( |
|
|
"--large", |
|
|
action="store_true", |
|
|
help="Upload using upload-large-folder", |
|
|
) |
|
|
parser.add_argument( |
|
|
"--token", |
|
|
type=str, |
|
|
default=HF_TOKEN, |
|
|
dest="HF_TOKEN", |
|
|
help=f"Your access token.", |
|
|
) |
|
|
args = parser.parse_args() |
|
|
if args.large: |
|
|
mt.push_large_folder_to_hf(repo_id, repo_type) |
|
|
else: |
|
|
mt.push_to_hf(repo_id, repo_type) |
|
|
|
