rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Merge branch 'henk717:united' into united

Browse Source
This commit is contained in:
ebolam
2022-03-05 13:25:10 -05:00
committed by GitHub
parent 4dd119c38d c20435855b
commit 4a8d7f5e0b
1 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@ -1015,9 +1015,10 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme
import transformers.generation_utils import transformers.generation_utils
from transformers import __version__ as transformers_version from transformers import __version__ as transformers_version
# Temporary fix for XGLM positional embedding issues until # Some versions of transformers 4.17.0.dev0 are affected by
# https://github.com/huggingface/transformers/issues/15736 # https://github.com/huggingface/transformers/issues/15736
# is resolved # This is a workaround for those versions of transformers.
if(transformers_version == "4.17.0.dev0"):
try: try:
from transformers.models.xglm.modeling_xglm import XGLMSinusoidalPositionalEmbedding from transformers.models.xglm.modeling_xglm import XGLMSinusoidalPositionalEmbedding
except ImportError: except ImportError: