fix ci

yuanlehome · yuanlehome · commit e358ce3246e3 · 2024-07-31T04:04:36.000Z
diff --git a/paddlenlp/experimental/transformers/fused_transformer_layers.py b/paddlenlp/experimental/transformers/fused_transformer_layers.py
@@ -1000,7 +1000,7 @@ def init_weight_shape(self, config):
         )
         self.ffn2_weight_shape = [self.embed_dim, self.dim_feedforward]
 
-        if self.quant_type == "weight_only_int4":
+        if config.quant_type == "weight_only_int4":
             self.qkv_weight_shape[0] //= 2
             self.linear_weight_shape[0] //= 2
             self.ffn1_weight_shape[0] //= 2

Original file line number	Diff line number	Diff line change
`@@ -1000,7 +1000,7 @@ def init_weight_shape(self, config):`
`1000`	`1000`	`)`
`1001`	`1001`	`self.ffn2_weight_shape = [self.embed_dim, self.dim_feedforward]`
`1002`	`1002`
`1003`		`- if self.quant_type == "weight_only_int4":`
	`1003`	`+ if config.quant_type == "weight_only_int4":`
`1004`	`1004`	`self.qkv_weight_shape[0] //= 2`
`1005`	`1005`	`self.linear_weight_shape[0] //= 2`
`1006`	`1006`	`self.ffn1_weight_shape[0] //= 2`