compatible for chatglm

Author: qc903113684

mlc_llm/core.py

@@ -627,6 +627,12 @@ def build_model_from_args(args: argparse.Namespace):
                     temperature=1.2,
                     repetition_penalty=0.996,
                 )
+            elif args.model_category == "chatglm":
+                dump_mlc_chat_config(
+                    args,
+                    vocab_size=config["padded_vocab_size"],
+                    max_window_size=model_config.max_sequence_length,
+                )
             else:
                 dump_mlc_chat_config(
                     args,