Update upload_directly_to_hf.py

upload_directly_to_hf.py

@@ -1,21 +1,19 @@
 #!/usr/bin/env python3
 """
 Create and upload SafeTensors DIRECTLY to Hugging Face
-No local folder needed - goes straight to your repo!
+Uses secure login - NO TOKEN IN CODE!
 """
 
 import torch
 import json
 from safetensors.torch import save_file
-from huggingface_hub import HfApi, create_repo
+from huggingface_hub import HfApi
 import tempfile
-import math
 from pathlib import Path
 
-# ============ CONFIGURE THIS ============
-REPO_NAME = "your-username/mineral-nano-1"  # CHANGE THIS to your HF username!
-HF_TOKEN = "your_token_here"  # Get from huggingface.co/settings/tokens
-# ========================================
+# ============ CONFIGURE ONLY THIS ============
+REPO_NAME = "Luke-Bergen/mineral-nano-1"  # CHANGE THIS to your HF username!
+# =============================================
 
 def initialize_weights(shape, init_type="normal", std=0.02):
     """Initialize tensor with proper initialization"""
@@ -164,15 +162,15 @@ def create_mineral_nano_weights():
     
     return state_dict
 
-def upload_to_huggingface(state_dict, repo_name, token):
-    """Upload SafeTensors directly to Hugging Face"""
+def upload_to_huggingface(state_dict, repo_name):
+    """Upload SafeTensors directly to Hugging Face using saved credentials"""
     
     print(f"\n{'='*60}")
     print(f"Uploading to Hugging Face: {repo_name}")
     print(f"{'='*60}")
     
-    # Initialize HF API
-    api = HfApi(token=token)
+    # Initialize HF API (uses saved token from huggingface-cli login)
+    api = HfApi()
     
     # Create temporary directory for files
     with tempfile.TemporaryDirectory() as temp_dir:
@@ -268,16 +266,19 @@ if __name__ == "__main__":
     print("="*60)
     
     # Verify configuration
-    if "your-username" in REPO_NAME or "your_token" in HF_TOKEN:
+    if "your-username" in REPO_NAME:
         print("\n❌ ERROR: Please configure the script first!")
-        print("\nEdit these lines at the top of the script:")
+        print("\nEdit this line at the top of the script:")
         print(f'  REPO_NAME = "your-username/mineral-nano-1"')
-        print(f'  HF_TOKEN = "your_token_here"')
-        print("\nGet your token from: https://huggingface.co/settings/tokens")
+        print("\nChange 'your-username' to YOUR HuggingFace username")
+        print("\nThen run: huggingface-cli login")
+        print("(Paste your token when prompted)")
         exit(1)
     
     print(f"\nTarget repository: {REPO_NAME}")
     print("This will take 10-20 minutes...")
+    print("\nMake sure you ran: huggingface-cli login")
+    input("\nPress ENTER to continue or Ctrl+C to cancel...")
     
     # Create weights
     print("\n" + "="*60)
@@ -289,6 +290,6 @@ if __name__ == "__main__":
     print("\n" + "="*60)
     print("STEP 2: Uploading to Hugging Face")
     print("="*60)
-    upload_to_huggingface(state_dict, REPO_NAME, HF_TOKEN)
+    upload_to_huggingface(state_dict, REPO_NAME)
     
     print("\n✅ All done! Your model is live on Hugging Face!")