Remove comments

Xilinx · Jul 18, 2023 · d32fca4 · d32fca4
1 parent 06dc443
commit d32fca4
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/src/brevitas_examples/llm/llm_quant/mha_layers.py b/src/brevitas_examples/llm/llm_quant/mha_layers.py
@@ -296,11 +296,5 @@ def set_weight(value):
                 del state_dict[name]
         state_dict[prefix + 'mha.out_proj.weight'] = torch.eye(self.mha.out_proj.weight.shape[0])
         state_dict[prefix + 'mha.out_proj.bias'] = torch.zeros(self.mha.out_proj.bias.shape)
-        # elif prefix + 'self.output.dense.weight' in name:
-        #     state_dict[prefix + 'mha.out_proj.weight'] = value
-        #     del state_dict[name]
-        # elif prefix + 'self.output.dense.bias' in name:
-        #     state_dict[prefix + 'mha.out_proj.bias'] = value
-        #     del state_dict[name]
         return super()._load_from_state_dict(
             state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)