Upload train_ministral_n8n.py with huggingface_hub
Browse files- 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()
|
| 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(project="ministral-3b-n8n-sft")
|
| 37 |
|
| 38 |
print(f"Loading tokenizer from {MODEL_NAME}...")
|
| 39 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
|