Add missing import

This commit is contained in:
oobabooga 2023-09-18 16:26:54 -07:00
parent 605ec3c9f2
commit ff5d3d2d09

View File

@ -11,6 +11,7 @@ from exllamav2 import (
from exllamav2.generator import ExLlamaV2BaseGenerator, ExLlamaV2Sampler from exllamav2.generator import ExLlamaV2BaseGenerator, ExLlamaV2Sampler
from modules import shared from modules import shared
from modules.logging_colors import logger
from modules.text_generation import get_max_prompt_length from modules.text_generation import get_max_prompt_length
try: try: