UMA-IA/CENTAURUS-Components-v2 est une version spécialisée du modèle Mistral-7B, fine-tunée pour fournir des réponses précises et concises aux questions techniques concernant les composants aérospatiaux et aéronautiques. Le modèle s'appuie sur le dataset UMA-IA/VELA-Components-v2 pour améliorer sa compréhension des composants de propulsion, leurs caractéristiques techniques, et leur maintenance.
Ce modèle a été fine-tuné sur le dataset UMA-IA/VELA-Components-v2, qui contient environ 800 paires question-réponse concernant divers composants aérospatiaux. Le processus de fine-tuning a utilisé la technique LoRA (Low-Rank Adaptation) pour adapter efficacement le modèle Mistral-7B à ce domaine spécifique.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Charger le modèle et le tokenizer
4model_name = "UMA-IA/CENTAURUS-Components-v2"
5model = AutoModelForCausalLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8# Format d'entrée recommandé
9question = "Quelle est la durée de vie moyenne d'une tuyère?"
10context = "Type: COMPOSANT, Composant: Tuyère, Catégorie: DURÉE_DE_VIE, Thème: question_simple"
11input_text = f"Question: {question}\nContexte: {context}\nRéponse:"
12
13# Générer une réponse
14inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
15outputs = model.generate(
16 inputs["input_ids"],
17 max_new_tokens=50,
18 temperature=0.7,
19 top_p=0.9
20)
21response = tokenizer.decode(outputs[0], skip_special_tokens=True)
22print(response)
Question: Quelle est la durée de vie moyenne d'une tuyère?
Contexte: Type: COMPOSANT, Composant: Tuyère, Catégorie: DURÉE_DE_VIE, Thème: question_simple
Réponse:
La durée de vie moyenne d'une tuyère est de 1500 à 2000 cycles d'utilisation. Les températures extrêmes et l'érosion thermique sont les principaux facteurs limitants.
