File size: 542 Bytes
2d05729 54b8ba7 2d05729 54b8ba7 2d05729 54b8ba7 2d05729 54b8ba7 2d05729 54b8ba7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# Example: Load and Use the Model
import torch
import json
from huggingface_hub import hf_hub_download
# Your repository ID
repo_id = "YOUR_USERNAME/YOUR_REPO_NAME" # Update this!
# Download files
config_path = hf_hub_download(repo_id=repo_id, filename="model_config.json")
weights_path = hf_hub_download(repo_id=repo_id, filename="model_weights.pt")
tokenizer_path = hf_hub_download(repo_id=repo_id, filename="tokenizer.json")
print("Files downloaded successfully!")
# Load and use your model
# (Add your TransformerModel class here)
|