cleanup code

ModelCloud · Jun 28, 2024 · f37b4ae · f37b4ae
1 parent 8670dc4
commit f37b4ae
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/gptqmodel/nn_modules/qlinear/qlinear_exllama.py b/gptqmodel/nn_modules/qlinear/qlinear_exllama.py
@@ -55,15 +55,9 @@ def __init__(self, bits: int, group_size: int , sym:bool, desc_act: bool, infeat
         self.original_outfeatures = outfeatures
         self.original_infeatures = infeatures
 
-        # code bug prevention
-        del infeatures
-        del outfeatures
-        del group_size
-
         self.maxq = 2**self.bits - 1
 
         assert self.infeatures % 32 == 0
-        assert self.infeatures % self.group_size == 0
         assert self.outfeatures % 32 == 0
 
         self.register_buffer(

diff --git a/gptqmodel/nn_modules/qlinear/qlinear_exllamav2.py b/gptqmodel/nn_modules/qlinear/qlinear_exllamav2.py
@@ -118,16 +118,9 @@ def __init__(self, bits: int, group_size: int, sym: bool, desc_act: bool, infeat
         # backup original values
         self.original_outfeatures = outfeatures
         self.original_infeatures = infeatures
-
-        # code bug prevention
-        del infeatures
-        del outfeatures
-        del group_size
-
         self.maxq = 2**self.bits - 1
 
         assert self.infeatures % 32 == 0
-        assert self.infeatures % self.group_size == 0
         assert self.outfeatures % 32 == 0
 
         # I need to register the tensors, otherwise, we won't be able to load them easily using transformers ...