Update web_demo.py

pull/637/head
zhengjie.xu 2024-01-19 19:01:32 +08:00 committed by GitHub
parent ad4a918402
commit 33b87bcba9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions

View File

@ -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__)