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

Fix import

Browse Source
This commit is contained in:
somebody
2023-07-28 16:03:55 -05:00
parent 184c3d9302
commit eb95c45ff1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -71,7 +71,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForToken
import transformers import transformers
import ipaddress import ipaddress
from functools import wraps from functools import wraps
from modeling.inference_models.utils import RestrictedUnpickler, use_custom_unpickler from modeling.pickling import RestrictedUnpickler, use_custom_unpickler
try: try:
from transformers.models.opt.modeling_opt import OPTDecoder from transformers.models.opt.modeling_opt import OPTDecoder