fix group size input in build text model

text.py

@@ -164,19 +164,19 @@ def build_text_model(config: TextConfig, dtype: torch.dtype = torch.float16) ->
                             "ln": nn.LayerNorm(config.dim, dtype=dtype),
                             "attn": nn.ModuleDict(
                                 {
-                                    "qkv": QuantizedLinear(config.dim, qkv_dim, group_size=config.text.group_size, dtype=dtype),
+                                    "qkv": QuantizedLinear(config.dim, qkv_dim, group_size=config.group_size, dtype=dtype),
                                     "proj": QuantizedLinear(
-                                        config.dim, config.dim, group_size=config.text.group_size, dtype=dtype
+                                        config.dim, config.dim, group_size=config.group_size, dtype=dtype
                                     ),
                                 }
                             ),
                             "mlp": nn.ModuleDict(
                                 {
                                     "fc1": QuantizedLinear(
-                                        config.dim, config.ff_dim, group_size=config.text.group_size, dtype=dtype
+                                        config.dim, config.ff_dim, group_size=config.group_size, dtype=dtype
                                     ),
                                     "fc2": QuantizedLinear(
-                                        config.ff_dim, config.dim, group_size=config.text.group_size, dtype=dtype
+                                        config.ff_dim, config.dim, group_size=config.group_size, dtype=dtype
                                     ),
                                 }
                             ),