From ff5d3d2d092af726edbfd93ef774578416c4b2d2 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Mon, 18 Sep 2023 16:26:54 -0700 Subject: [PATCH] Add missing import --- modules/exllamav2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/exllamav2.py b/modules/exllamav2.py index 2758fa2d..605a0927 100644 --- a/modules/exllamav2.py +++ b/modules/exllamav2.py @@ -11,6 +11,7 @@ from exllamav2 import ( from exllamav2.generator import ExLlamaV2BaseGenerator, ExLlamaV2Sampler from modules import shared +from modules.logging_colors import logger from modules.text_generation import get_max_prompt_length try: