stmasson commited on
Commit
6d492a0
·
verified ·
1 Parent(s): a779a89

Upload scripts/train_n8n_sft.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- wandb_key = os.environ.get("WANDB_API_KEY")
90
- if wandb_key:
91
- import wandb
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