remove test, fix default in order to fix tests

vllm-project · Dec 10, 2024 · ccb007f · ccb007f
1 parent 146e4be
commit ccb007f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 42 deletions.
diff --git a/src/llmcompressor/modifiers/quantization/gptq/base.py b/src/llmcompressor/modifiers/quantization/gptq/base.py
@@ -120,7 +120,7 @@ class GPTQModifier(Modifier, HooksMixin):
     disable_quantization_observer_epoch: Optional[float] = None
 
     # private variables
-    _quantization_modifier: Optional[QuantizationModifier] = PrivateAttr()
+    _quantization_modifier: Optional[QuantizationModifier] = PrivateAttr(default=None)
     _hessians: Dict[torch.nn.Module, torch.Tensor] = PrivateAttr(default_factory=dict)
     _num_samples: Dict[torch.nn.Module, int] = PrivateAttr(default_factory=dict)
     _update_size: Optional[int] = PrivateAttr(default=None)

diff --git a/tests/llmcompressor/modifiers/quantization/gptq/utils/test_gptq_wrapper.py b/tests/llmcompressor/modifiers/quantization/gptq/utils/test_gptq_wrapper.py