amkkk's picture
Upload checkpoint-style abliteration model
402d5fa verified
raw
history blame contribute delete
1.09 kB
import argparse
import os
from huggingface_hub import HfApi
def main():
parser = argparse.ArgumentParser(description="Upload a prepared model folder to Hugging Face Hub.")
parser.add_argument("--repo-id", type=str, required=True, help="e.g. username/qwen3.5-0.8b-abliterated")
parser.add_argument("--folder", type=str, default=".", help="Local folder to upload")
parser.add_argument("--private", action="store_true", help="Create private repository")
parser.add_argument("--token", type=str, default=os.getenv("HF_TOKEN"))
args = parser.parse_args()
api = HfApi(token=args.token)
api.create_repo(repo_id=args.repo_id, repo_type="model", private=args.private, exist_ok=True)
api.upload_folder(
repo_id=args.repo_id,
repo_type="model",
folder_path=args.folder,
ignore_patterns=[".cache/*", "**/.cache/*", "*.lock"],
commit_message="Upload checkpoint-style abliteration model",
)
print(f"Uploaded: https://huggingface.co/{args.repo_id}")
if __name__ == "__main__":
main()