Upload scripts/train_n8n_sft.py with huggingface_hub
Browse files- scripts/train_n8n_sft.py +4 -10
scripts/train_n8n_sft.py
CHANGED
|
@@ -7,13 +7,13 @@
|
|
| 7 |
# "datasets>=3.0.0",
|
| 8 |
# "accelerate>=1.0.0",
|
| 9 |
# "bitsandbytes>=0.44.0",
|
| 10 |
-
# "wandb>=0.18.0",
|
| 11 |
# "huggingface_hub>=0.26.0",
|
| 12 |
# "torch>=2.4.0",
|
| 13 |
# "einops>=0.8.0",
|
| 14 |
# "sentencepiece>=0.2.0",
|
| 15 |
# ]
|
| 16 |
# [tool.uv]
|
|
|
|
| 17 |
# extra-index-url = ["https://download.pytorch.org/whl/cu124"]
|
| 18 |
# ///
|
| 19 |
"""
|
|
@@ -86,15 +86,9 @@ if hf_token:
|
|
| 86 |
else:
|
| 87 |
print("Warning: HF_TOKEN non défini, push désactivé")
|
| 88 |
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
wandb.login(key=wandb_key)
|
| 93 |
-
report_to = "wandb"
|
| 94 |
-
print("Tracking W&B activé")
|
| 95 |
-
else:
|
| 96 |
-
report_to = "none"
|
| 97 |
-
print("Tracking W&B désactivé")
|
| 98 |
|
| 99 |
# ============================================================================
|
| 100 |
# CHARGEMENT DU MODÈLE
|
|
|
|
| 7 |
# "datasets>=3.0.0",
|
| 8 |
# "accelerate>=1.0.0",
|
| 9 |
# "bitsandbytes>=0.44.0",
|
|
|
|
| 10 |
# "huggingface_hub>=0.26.0",
|
| 11 |
# "torch>=2.4.0",
|
| 12 |
# "einops>=0.8.0",
|
| 13 |
# "sentencepiece>=0.2.0",
|
| 14 |
# ]
|
| 15 |
# [tool.uv]
|
| 16 |
+
# index-strategy = "unsafe-best-match"
|
| 17 |
# extra-index-url = ["https://download.pytorch.org/whl/cu124"]
|
| 18 |
# ///
|
| 19 |
"""
|
|
|
|
| 86 |
else:
|
| 87 |
print("Warning: HF_TOKEN non défini, push désactivé")
|
| 88 |
|
| 89 |
+
# Désactivé wandb pour éviter les conflits de dépendances
|
| 90 |
+
report_to = "none"
|
| 91 |
+
print("Tracking désactivé (pas de wandb)")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 92 |
|
| 93 |
# ============================================================================
|
| 94 |
# CHARGEMENT DU MODÈLE
|