diff --git a/agent/pal_inference.py b/agent/pal_inference.py index a885905..ed55390 100644 --- a/agent/pal_inference.py +++ b/agent/pal_inference.py @@ -17,6 +17,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +# isort: skip_file import argparse import copy import json diff --git a/chat/web_demo.py b/chat/web_demo.py index 20f6eff..478043c 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -14,6 +14,7 @@ Please run with the command `streamlit run path/to/web_demo.py --server.address=0.0.0.0 --server.port 7860`. Using `python path/to/web_demo.py` may cause unknown problems. """ +# isort: skip_file import copy import warnings from dataclasses import asdict, dataclass @@ -22,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