from transformers import AutoModelForCausalLM
import torch

def load_model(model_name: str):
    print(f"🧠 Chargement du modèle {model_name}...")
    model = AutoModelForCausalLM.from_pretrained(
        model_name,
        torch_dtype=torch.bfloat16
    )
    return model