add version check

wanli2014 · Jun 30, 2023 · 5810278 · 5810278
1 parent e8d18a1
commit 5810278
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/utils/common.py b/src/utils/common.py
@@ -231,6 +231,7 @@ def load_pretrained(
         output_embedding_base_layer = model.transformer
         output_embedding_layer_name = "output_layer"
     else:
+        assert tokenizer.eos_token_id == 150005, "Please specify `use_v2` argument while using ChatGLM2-6B."
         output_embedding_base_layer = model
         output_embedding_layer_name = "lm_head"