diff --git a/chat/web_demo.py b/chat/web_demo.py index 9ba74a6..7e76e9f 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -17,8 +17,9 @@ from typing import Callable, List, Optional import streamlit as st import torch from torch import nn -from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip -from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList +from transformers import AutoModelForCausalLM, AutoTokenizer +from transformers.generation.utils import (LogitsProcessorList, + StoppingCriteriaList) from transformers.utils import logging logger = logging.get_logger(__name__)