diff --git a/extensions/openai/completions.py b/extensions/openai/completions.py index ce9cc102..e100472c 100644 --- a/extensions/openai/completions.py +++ b/extensions/openai/completions.py @@ -22,7 +22,12 @@ from modules.chat import ( load_instruction_template_memoized ) from modules.presets import load_preset_memoized -from modules.text_generation import decode, encode, generate_reply, get_reply_from_output_ids +from modules.text_generation import ( + decode, + encode, + generate_reply, + get_reply_from_output_ids +) class LogitsBiasProcessor(LogitsProcessor): diff --git a/modules/html_generator.py b/modules/html_generator.py index 71a61bd0..e3dd453e 100644 --- a/modules/html_generator.py +++ b/modules/html_generator.py @@ -7,8 +7,8 @@ from pathlib import Path import markdown from PIL import Image, ImageOps -from modules.utils import get_available_chat_styles from modules import shared +from modules.utils import get_available_chat_styles # This is to store the paths to the thumbnails of the profile pictures image_cache = {}