From aad73667af6a1dd47be38f680ebf5bd78294b6b9 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Mon, 22 Jan 2024 03:25:55 -0800 Subject: [PATCH] Lint --- extensions/openai/completions.py | 7 ++++++- modules/html_generator.py | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) 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 = {}