Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Charger le modèle
4model = AutoModelForCausalLM.from_pretrained(
5 "BelikanM/kibali-final-merged",
6 torch_dtype="auto",
7 device_map="auto"
8)
9tokenizer = AutoTokenizer.from_pretrained("BelikanM/kibali-final-merged")
10
11# Exemple d'utilisation
12prompt = "[INST] Explique-moi le principe de la tomographie électrique [/INST]"
13inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
14outputs = model.generate(**inputs, max_new_tokens=200)
15response = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(response)[INST] Comment interpréter une tomographie de résistivité électrique ? [/INST][INST] Quelles sont les limites du GPR en milieu urbain ? [/INST][INST] Applications de la magnétométrie en archéologie [/INST]