From 33b87bcba9fac56070bf2c2137b09d32016cbc85 Mon Sep 17 00:00:00 2001 From: "zhengjie.xu" Date: Fri, 19 Jan 2024 19:01:32 +0800 Subject: [PATCH] Update web_demo.py --- chat/web_demo.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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__)