diff --git a/aiserver.py b/aiserver.py index fd8f00fa..194fa779 100644 --- a/aiserver.py +++ b/aiserver.py @@ -1234,7 +1234,7 @@ def patch_transformers(): # Fix a bug in OPTForCausalLM where self.lm_head is the wrong size - if(packaging.version.parse("4.19.0.dev0") <= packaging.version.parse(transformers_version) <= packaging.version.parse("4.19.2")): + if(packaging.version.parse("4.19.0.dev0") <= packaging.version.parse(transformers_version) < packaging.version.parse("4.20.0")): try: from transformers import OPTForCausalLM, OPTModel except ImportError: