alert pad_token source

src/llmtuner/extras/template.py

@@ -204,6 +204,9 @@ def get_template_and_fix_tokenizer(
         logger.info("Add eos token: {}".format(tokenizer.eos_token))
 
     if tokenizer.pad_token_id is None:
+        if tokenizer.unk_token_id is not None:
+            tokenizer.pad_token = tokenizer.unk_token
+        else:
             tokenizer.pad_token = tokenizer.eos_token
         logger.info("Add pad token: {}".format(tokenizer.pad_token))
 

src/llmtuner/tuner/core/loader.py

@@ -154,7 +154,7 @@ def load_model_and_tokenizer(
         model.generate = MethodType(PreTrainedModel.generate, model)
 
     # Fix LM head (for ChatGLM2)
-    if not hasattr(model, "lm_head"):
+    if not hasattr(model, "lm_head") and hasattr(model, "transformer"):
         setattr(model, "lm_head", model.transformer.output_layer)
 
     # Register auto class to save the custom code files.