Upload train_aviation.py with huggingface_hub
Browse files- train_aviation.py +15 -1
train_aviation.py
CHANGED
|
@@ -18,7 +18,21 @@ import os
|
|
| 18 |
from huggingface_hub import list_repo_files
|
| 19 |
|
| 20 |
# DEBUG: Check token and repo access
|
| 21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
from datasets import load_dataset
|
| 24 |
from peft import LoraConfig, prepare_model_for_kbit_training, get_peft_model
|
|
|
|
| 18 |
from huggingface_hub import list_repo_files
|
| 19 |
|
| 20 |
# DEBUG: Check token and repo access
|
| 21 |
+
print("🔍 DIAGNOSTICS:")
|
| 22 |
+
token = os.environ.get("HF_TOKEN")
|
| 23 |
+
print(f" HF_TOKEN env var present: {bool(token)}")
|
| 24 |
+
if token:
|
| 25 |
+
print(f" HF_TOKEN prefix: {token[:4]}...")
|
| 26 |
+
|
| 27 |
+
model_id = "mistralai/Ministral-3-14B-Reasoning-2512"
|
| 28 |
+
try:
|
| 29 |
+
print(f" Attempting to list files for {model_id}...")
|
| 30 |
+
files = list_repo_files(model_id, token=token)
|
| 31 |
+
print(f" ✅ Success! Found {len(files)} files.")
|
| 32 |
+
print(f" First 5 files: {files[:5]}")
|
| 33 |
+
except Exception as e:
|
| 34 |
+
print(f" ❌ Failed to list repo files: {e}")
|
| 35 |
+
print("="*40)
|
| 36 |
|
| 37 |
from datasets import load_dataset
|
| 38 |
from peft import LoraConfig, prepare_model_for_kbit_training, get_peft_model
|