From 4c022ff0e86768a44b7d11ab4cc76c4f1c18c2eb Mon Sep 17 00:00:00 2001 From: Liqu1d-G <1643076551@qq.com> Date: Fri, 26 Jan 2024 17:03:32 +0800 Subject: [PATCH] [CI]: fix flake8 error --- chat/web_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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