import subprocess | |
import os | |
# Define the paths | |
converter_path = './zero_to_fp32.py' | |
huggingface_repo = 'https://huggingface.co/TannerGladson/chess-roberta-whole-move-pretrained' | |
# Extract the model | |
subprocess.run(['python', converter_path, ".", "pytorch_model.bin"], check=True) | |
# Commit and push | |
# Check if the current directory is a git repository | |
if not os.path.exists('.git'): | |
result = subprocess.run(['git', 'clone', huggingface_repo, '.']) | |
if result.returncode != 0: | |
raise RuntimeError("Failed to clone the repository.") | |
subprocess.run(['git', 'lfs', 'install'], check=True) | |
subprocess.run(['git', 'add', '.'], check=True) | |
subprocess.run(['git', 'commit', '-m', 'Run zero_to_fp32.py and commit changes'], check=True) | |
subprocess.run(['git', 'push'], check=True) |