diff --git a/aiserver.py b/aiserver.py index 9b9a1c76..05fa7dae 100644 --- a/aiserver.py +++ b/aiserver.py @@ -1242,7 +1242,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: