Commit 3c35c466 authored by zhuwenwen's avatar zhuwenwen
Browse files

fix indentation of self.quant_method

parent df704163
...@@ -209,10 +209,10 @@ class BaiChuanAttention(nn.Module): ...@@ -209,10 +209,10 @@ class BaiChuanAttention(nn.Module):
quant_config=quant_config, quant_config=quant_config,
prefix=f"{prefix}.attn", prefix=f"{prefix}.attn",
) )
self.quant_method = None self.quant_method = None
if quant_config is not None: if quant_config is not None:
self.quant_method=quant_config.get_name() self.quant_method=quant_config.get_name()
self.quant_config=quant_config self.quant_config=quant_config
def forward( def forward(
self, self,
......
...@@ -210,7 +210,7 @@ class FalconAttention(nn.Module): ...@@ -210,7 +210,7 @@ class FalconAttention(nn.Module):
quant_config=quant_config, quant_config=quant_config,
prefix=f"{prefix}.attn", prefix=f"{prefix}.attn",
) )
self.quant_method = None self.quant_method = None
if quant_config is not None: if quant_config is not None:
self.quant_method=quant_config.get_name() self.quant_method=quant_config.get_name()
self.quant_config=quant_config self.quant_config=quant_config
......
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