diff --git a/chat/web_demo.py b/chat/web_demo.py index 7e76e9f..18af3c6 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -8,7 +8,6 @@ Please refer to these links below for more information: Please run with the command `streamlit run path/to/web_demo.py --server.address=0.0.0.0 --server.port 7860`. Using `python path/to/web_demo.py` may cause unknown problems. """ - import copy import warnings from dataclasses import asdict, dataclass @@ -17,11 +16,13 @@ from typing import Callable, List, Optional import streamlit as st import torch from torch import nn -from transformers import AutoModelForCausalLM, AutoTokenizer from transformers.generation.utils import (LogitsProcessorList, StoppingCriteriaList) from transformers.utils import logging +from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip + + logger = logging.get_logger(__name__)