rm references to fs2 modules

pytorch · Apr 5, 2024 · 4256bbd · 4256bbd
1 parent 6efbd08
commit 4256bbd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/quantize.py b/quantize.py
@@ -234,7 +234,7 @@ def create_quantized_state_dict(self) -> Dict:
 
         for fqn, mod in self.mod.named_modules():
             # print(f"maybe? quantize {fqn}...{type(mod)}")
-            if isinstance(mod, torch.nn.Linear) or isinstance(mod, fsLinear):
+            if isinstance(mod, torch.nn.Linear):
                 # print(f"candidate {fqn}, nodetype {self.node_type}")
                 if (
                     (self.node_type == "*")
@@ -371,8 +371,6 @@ def create_quantized_state_dict(self) -> Dict:
         for fqn, mod in self.mod.named_modules():
             if (
                 isinstance(mod, nn.Embedding)
-                or isinstance(mod, fsEmbedding)
-                or isinstance(mod, fsStandardEmbedding)
             ):
                 # print("****")
                 # print(f"Embedding identified: {fqn, mod}")