From 86c5da6dd516f88fbebff99a0fe9eeadbf1301ae Mon Sep 17 00:00:00 2001 From: "zhengjie.xu" Date: Fri, 19 Jan 2024 17:16:08 +0800 Subject: [PATCH] fix lint --- 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 99e9c0d..9ba74a6 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -17,7 +17,7 @@ from typing import Callable, List, Optional import streamlit as st import torch from torch import nn -from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip +from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList from transformers.utils import logging