Add model_name attribute to model classes

outlines/models/exllamav2.py

@@ -42,6 +42,8 @@ def decode(self, token_ids: "torch.LongTensor") -> List[str]:
 class ExLlamaV2Model:
     """Represents a `exl2` model."""
 
+    model_name = "ExLlamaV2"
+
     def __init__(
         self,
         generator: "ExLlamaV2DynamicGenerator",

outlines/models/llamacpp.py

@@ -138,6 +138,8 @@ class LlamaCpp:
 
     """
 
+    model_name = "LlamaCpp"
+
     def __init__(self, model: "Llama"):
         self.model = model
 

outlines/models/mlxlm.py

@@ -17,6 +17,8 @@ class MLXLM:
     Represents an `mlx_lm` model
     """
 
+    model_name = "MLXLM"
+
     def __init__(
         self,
         model: "nn.Module",

outlines/models/transformers.py

@@ -129,6 +129,8 @@ def __setstate__(self, state):
 class Transformers:
     """Represents a `transformers` model."""
 
+    model_name = "Transformers"
+
     def __init__(
         self,
         model: "PreTrainedModel",

outlines/models/vllm.py

@@ -19,6 +19,8 @@ class VLLM:
 
     """
 
+    model_name = "VLLM"
+
     def __init__(self, model: "LLM"):
         self.model = model
         self.lora_request = None