fix #4917

hiyouga · Jul 22, 2024 · 26082fc · 26082fc
1 parent c333e2f
commit 26082fc
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/llamafactory/model/patcher.py b/src/llamafactory/model/patcher.py
@@ -89,9 +89,6 @@ def patch_config(
     if getattr(config, "model_type", None) == "qwen2" and is_trainable and model_args.flash_attn == "fa2":
         setattr(config, "use_cache", False)  # qwen2 does not support use_cache when using flash attn
 
-    if getattr(config, "model_type", None) == "chatglm":
-        require_version("transformers==4.41.2", "To fix: pip install transformers==4.41.2")
-
     # deepspeed zero3 is not compatible with low_cpu_mem_usage
     init_kwargs["low_cpu_mem_usage"] = model_args.low_cpu_mem_usage and (not is_deepspeed_zero3_enabled())