Remove task map as each model would have a default task

Author: XkunW

vec_inf/client/_client_vars.py

@@ -10,8 +10,6 @@
     Signature string indicating successful model server startup
 SRC_DIR : str
     Absolute path to the package source directory
-VLLM_TASK_MAP : dict
-    Mapping of model types to their corresponding vLLM tasks
 REQUIRED_FIELDS : set
     Set of required fields for model configuration
 KEY_METRICS : dict
@@ -30,14 +28,6 @@
 SRC_DIR = str(Path(__file__).parent.parent)
 
 
-# Maps model types to vLLM tasks
-VLLM_TASK_MAP = {
-    "LLM": "generate",
-    "VLM": "generate",
-    "TEXT_EMBEDDING": "embed",
-    "REWARD_MODELING": "reward",
-}
-
 # Required fields for model configuration
 REQUIRED_FIELDS = {
     "model_family",

vec_inf/client/_slurm_script_generator.py

@@ -11,7 +11,6 @@
 from vec_inf.client._client_vars import (
     SLURM_JOB_CONFIG_ARGS,
     SLURM_SCRIPT_TEMPLATE,
-    VLLM_TASK_MAP,
 )
 
 
@@ -43,7 +42,6 @@ def __init__(self, params: dict[str, Any]):
         self.model_weights_path = str(
             Path(params["model_weights_parent_dir"], params["model_name"])
         )
-        self.task = VLLM_TASK_MAP[self.params["model_type"]]
 
     def _generate_script_content(self) -> str:
         """Generate the complete SLURM script content.