update weight map

InternLM · Dec 15, 2023 · 0935ddb · 0935ddb
1 parent f79e080
commit 0935ddb
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/lmdeploy/turbomind/deploy/target_model/base.py b/lmdeploy/turbomind/deploy/target_model/base.py
@@ -84,9 +84,7 @@ def valid(self):
 
 
 _WEIGHT_DTYPE_MAP = dict(
-    w4=torch.float16,
     int4=torch.float16,
-    half=torch.float16,
     fp16=torch.float16,
     fp32=torch.float16,
     bf16=torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float16,