Spaces:
Sleeping
Sleeping
File size: 618 Bytes
73c5b8e d808b10 73c5b8e 4765df3 73c5b8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from core.model_loader import load_model
from core.tokenizer_loader import load_tokenizer
from core.dataset_loader import load_dataset_single
dataset = load_dataset_single()
from core.trainer import build_trainer
MODEL_NAME = "tiiuae/falcon-7b-instruct"
HUB_ID = "Mauricio-100/agent-ai"
print("🚀 Lancement de l’entraînement de Gopu Agent...")
model = load_model(MODEL_NAME)
tokenizer = load_tokenizer(MODEL_NAME)
dataset = load_multiple_datasets()
trainer = build_trainer(model, dataset, tokenizer, HUB_ID)
trainer.train()
trainer.push_to_hub()
print("✅ Modèle entraîné et publié sur Hugging Face !")
|