This commit is contained in:
oobabooga 2024-01-22 03:25:55 -08:00
parent 6ada77cf5a
commit aad73667af
2 changed files with 7 additions and 2 deletions

View File

@ -22,7 +22,12 @@ from modules.chat import (
load_instruction_template_memoized load_instruction_template_memoized
) )
from modules.presets import load_preset_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): class LogitsBiasProcessor(LogitsProcessor):

View File

@ -7,8 +7,8 @@ from pathlib import Path
import markdown import markdown
from PIL import Image, ImageOps from PIL import Image, ImageOps
from modules.utils import get_available_chat_styles
from modules import shared 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 # This is to store the paths to the thumbnails of the profile pictures
image_cache = {} image_cache = {}