Skip to content

GitLab

Unverified Commit e1464c3a authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[Quantization] Enable compressed-tensors AWQ for Turing GPU (#29732) 


Signed-off-by: default avatarIsotr0py <mozf@mail2.sysu.edu.cn>
parent a491b091
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_wNa16.py
View file @ e1464c3a
...@@ -79,8 +79,8 @@ class CompressedTensorsWNA16(CompressedTensorsScheme): ...@@ -79,8 +79,8 @@ class CompressedTensorsWNA16(CompressedTensorsScheme):
@classmethod @classmethod
def get_min_capability(cls) -> int: def get_min_capability(cls) -> int:
# ampere and up # Turing and up
return 80 return 75
def create_weights( def create_weights(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment