fix isort

2024-01-19 18:58:02 +08:00 · 2024-01-19 18:58:02 +08:00 · ad4a918402
parent 86c5da6dd5
commit ad4a918402
1 changed files with 3 additions and 2 deletions
--- 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__)