stmasson commited on
Commit
1292a88
·
verified ·
1 Parent(s): 1ba34a4

Upload train_ministral_n8n.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_ministral_n8n.py +1 -1
train_ministral_n8n.py CHANGED
@@ -33,7 +33,7 @@ OUTPUT_MODEL = "stmasson/ministral-3b-n8n-workflows"
33
  MAX_SEQ_LENGTH = 4096 # n8n workflows can be long
34
 
35
  # Initialize Trackio for monitoring
36
- trackio.init(project_name="ministral-3b-n8n-sft")
37
 
38
  print(f"Loading tokenizer from {MODEL_NAME}...")
39
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
 
33
  MAX_SEQ_LENGTH = 4096 # n8n workflows can be long
34
 
35
  # Initialize Trackio for monitoring
36
+ trackio.init()
37
 
38
  print(f"Loading tokenizer from {MODEL_NAME}...")
39
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)