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

Cleanup

Browse Source
This commit is contained in:
somebody
2023-03-09 20:28:51 -06:00
parent 52095054a3
commit a472bdf6c3
1 changed files with 1 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
aiserver.py
View File

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