remove task validation not needed

huggingface · Jul 25, 2024 · d010916 · d010916
1 parent dbe1253
commit d010916
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/optimum_benchmark/backends/llama_cpp/config.py b/optimum_benchmark/backends/llama_cpp/config.py
@@ -26,9 +26,6 @@ class LlamaCppConfig(BackendConfig):
     def __post_init__(self):
         super().__post_init__()
 
-        if self.task not in TEXT_GENERATION_TASKS + TEXT_EMBEDDING_TASKS:
-            raise NotImplementedError(f"Llama.cpp does not support task {self.task}")
-
         self.device = self.device.lower()  # type: ignore
 
         if self.device not in ["cuda", "mps", "cpu"]: