Hugging Face's logo

stmasson
/
training-scripts

Model card Files
xet
Community
stmasson commited on
Commit
2d41996
·
verified ·
1 Parent(s): a4acc9b

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
@@ -27,7 +27,7 @@ from trl import SFTTrainer, SFTConfig
27
  import trackio
28
 
29
  # Configuration
30
- MODEL_NAME = "mistralai/Ministral-3b-instruct"
31
  DATASET_NAME = "stmasson/n8n-workflows-thinking"
32
  OUTPUT_MODEL = "stmasson/ministral-3b-n8n-workflows"
33
  MAX_SEQ_LENGTH = 4096 # n8n workflows can be long
 
27
  import trackio
28
 
29
  # Configuration
30
+ MODEL_NAME = "mistralai/Ministral-3-3B-Instruct-2512"
31
  DATASET_NAME = "stmasson/n8n-workflows-thinking"
32
  OUTPUT_MODEL = "stmasson/ministral-3b-n8n-workflows"
33
  MAX_SEQ_LENGTH = 4096 # n8n workflows can be long