Tree-AI-lab/kouman-identity-instruct
Viewer • Updated • 35k • 310 • 2
Kouman IA est un modèle d'intelligence artificielle souverain africain développé par Tree IA labs, spécialisé dans la langue baoulé (bci), le français (fr), le dioula (dyu) et l'anglais (en).
Le modèle intègre l'identité souveraine ivoirienne, des capacités de dialogue instruction-following, la programmation informatique, la traduction croisée et le raisonnement culturel.
Tree-AI-lab/kouman-traduction-culture-sft : 25,000 exemples de traduction et de culture ivoirienne.Tree-AI-lab/kouman-identity-instruct : 25,000 dialogues d'ancrage d'identité souveraine Kouman IA.Tree-AI-lab/kouman-situations-conseils-13k : 13,000 dialogues de situations et conseils pratiques.Tree-AI-lab/kouman-conversations-trilingues-longues-25k : 25,000 conversations longues de quatre échanges sur maths, culture, mythes et relations.from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
identifiant_modele = "Tree-AI-lab/kouman-IA-1.5-1B"
tokenizer = AutoTokenizer.from_pretrained(identifiant_modele)
modele = AutoModelForCausalLM.from_pretrained(
identifiant_modele,
torch_dtype=torch.bfloat16,
device_map="auto"
)
messages = [
{"role": "system", "content": "Tu es Kouman IA, l'assistant souverain africain créé par Tree IA labs."},
{"role": "user", "content": "N ti baoulé nun sran. Amun kan nian : ngue yɛle Kouman IA ?"}
]
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(prompt, return_tensors="pt").to(modele.device)
# Définition conjointe des tokens d'arrêt Llama 3.2
outputs = modele.generate(
**inputs,
max_new_tokens=256,
temperature=0.7,
eos_token_id=[tokenizer.eos_token_id, 128009]
)
print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))