rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-03-23 23:10:06 +01:00
Code Issues Projects Releases Wiki Activity

Merge branch 'united' into overhaul-merge

Browse Source
This commit is contained in:
Gnome Ann 2022-06-09 23:48:28 -04:00
commit ce582f188f
1 changed files with 1 additions and 1 deletions

2
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: