From ad4a918402d9295df7bea790646d33a1ae4bd830 Mon Sep 17 00:00:00 2001 From: "zhengjie.xu" Date: Fri, 19 Jan 2024 18:58:02 +0800 Subject: [PATCH] fix isort --- 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 9ba74a6..7e76e9f 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -17,8 +17,9 @@ 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.generation.utils import LogitsProcessorList, StoppingCriteriaList +from transformers import AutoModelForCausalLM, AutoTokenizer +from transformers.generation.utils import (LogitsProcessorList, + StoppingCriteriaList) from transformers.utils import logging logger = logging.get_logger(__name__)