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