diff --git a/chat/web_demo.py b/chat/web_demo.py index 478043c..5d37a2e 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -23,7 +23,7 @@ from typing import Callable, List, Optional import streamlit as st import torch from torch import nn -from transformers.generation.utils import (LogitsProcessorList, +from transformers.generation.utils import (LogitsProcessorList, StoppingCriteriaList) from transformers.utils import logging