remove a bit of unused code (#1536)

deepjavalibrary · Feb 12, 2024 · 3b91d39 · 3b91d39
1 parent 8bf54f7
commit 3b91d39
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/engines/python/setup/djl_python/properties_manager/hf_properties.py b/engines/python/setup/djl_python/properties_manager/hf_properties.py
@@ -19,21 +19,6 @@ class HFQuantizeMethods(str, Enum):
     bitsandbytes4 = 'bitsandbytes4'
     bitsandbytes8 = 'bitsandbytes8'
 
-    # TODO remove this after refactor of all handlers
-    # supported by vllm
-    awq = 'awq'
-
-
-LMI_DIST_ADV_MODEL = {
-    "RWForCausalLM",
-    "GPTNeoXForCausalLM",
-    "T5ForConditionalGeneration",
-    "LlamaForCausalLM",
-    "FalconForCausalLM",
-    "MPTForCausalLM",
-    "GPTBigCodeForCausalLM",
-}
-
 
 def get_torch_dtype_from_str(dtype: str):
     if dtype == "auto":