From eb95c45ff16bb0a729af298615a0ede3c746a89a Mon Sep 17 00:00:00 2001 From: somebody Date: Fri, 28 Jul 2023 16:03:55 -0500 Subject: [PATCH] Fix import --- aiserver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 8003e463..5c622793 100644 --- a/aiserver.py +++ b/aiserver.py @@ -71,7 +71,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForToken import transformers import ipaddress from functools import wraps -from modeling.inference_models.utils import RestrictedUnpickler, use_custom_unpickler +from modeling.pickling import RestrictedUnpickler, use_custom_unpickler try: from transformers.models.opt.modeling_opt import OPTDecoder