rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #143 from VE-FORBRYDERNE/overhaul-merge 

Merge united into overhaul
This commit is contained in:
henk717 2022-06-10 07:21:27 +02:00 committed by GitHub
parent 22e8468b98 ce582f188f
commit bd18cd6900
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -1234,7 +1234,7 @@ def patch_transformers():
# Fix a bug in OPTForCausalLM where self.lm_head is the wrong size # 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: try:
from transformers import OPTForCausalLM, OPTModel from transformers import OPTForCausalLM, OPTModel
except ImportError: except ImportError: