Create core/model_loader.py

core/model_loader.py

@@ -0,0 +1,10 @@
+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