UMA-IA/CENTAURUS-Engine-v1 is a specialized fine-tuned version of Mistral-7B designed to provide accurate and detailed answers to technical questions related to aerospace and aeronautical engines. The model leverages the UMA-IA/UMA_Dataset_Engine_Aero_LLM to enhance its understanding of complex engineering principles, propulsion systems, and aerospace technologies.
This model was fine-tuned on UMA-IA/VELA-Engine-v1, a curated dataset focusing on aerospace engines, propulsion systems, and general aeronautical engineering. The fine-tuning process was performed using supervised learning to adapt Mistral-7B to technical discussions.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "UMA-IA/CENTAURUS-Engine-v1"
4
5model = AutoModelForCausalLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8input_text = "Explain the working principle of a turbofan engine."
9inputs = tokenizer(input_text, return_tensors="pt")
10outputs = model.generate(**inputs)
11print(tokenizer.decode(outputs[0], skip_special_tokens=True))
