sunkencity commited on
Commit
31bab64
·
verified ·
1 Parent(s): 4d6ed12

Upload train_aviation.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- # ... (diagnostics code kept as is) ...
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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