diff --git a/aiserver.py b/aiserver.py index b6940e1e..7c3451c0 100644 --- a/aiserver.py +++ b/aiserver.py @@ -36,22 +36,16 @@ import ijson import datetime import collections import zipfile -import packaging import packaging.version -import contextlib import traceback import markdown import bleach -import itertools -import bisect import functools import traceback import inspect import warnings import multiprocessing -import copy import numpy as np -from collections.abc import Iterable from collections import OrderedDict from typing import Any, Callable, TypeVar, Tuple, Union, Dict, Set, List, Optional, Type @@ -70,13 +64,8 @@ from utils import debounce import utils import koboldai_settings import torch -from transformers import StoppingCriteria, GPT2Tokenizer, GPT2LMHeadModel, GPTNeoForCausalLM, GPTNeoModel, AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoTokenizer, PreTrainedModel, modeling_utils, AutoModelForTokenClassification -from transformers import __version__ as transformers_version +from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForTokenClassification import transformers -try: - from transformers.models.opt.modeling_opt import OPTDecoder -except: - pass # Text2img import base64