From 2a8bb2e7d996cb77e3c8179720d398b9a65e625b Mon Sep 17 00:00:00 2001 From: bmullick-amd Date: Tue, 17 Dec 2024 19:58:04 -0800 Subject: [PATCH] fix isort error --- vllm/model_executor/models/t5.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/vllm/model_executor/models/t5.py b/vllm/model_executor/models/t5.py index f981fe7606343..7957a2ac02ec7 100644 --- a/vllm/model_executor/models/t5.py +++ b/vllm/model_executor/models/t5.py @@ -19,9 +19,10 @@ import copy import os from typing import Iterable, List, Optional, Tuple -import torch # type: ignore -from torch import nn # type: ignore -import torch.nn.functional as F # type: ignore + +import torch # type: ignore +import torch.nn.functional as F # type: ignore +from torch import nn # type: ignore from transformers import T5Config from transformers.utils import logging @@ -67,8 +68,8 @@ def load_tf_weights_in_t5(model, config, tf_checkpoint_path): """Load tf checkpoints in a pytorch model.""" try: import re - import numpy as np # type: ignore - import tensorflow as tf # type: ignore + import numpy as np # type: ignore + import tensorflow as tf # type: ignore except ImportError: logger.error( "TensorFlow is to be installed. Please see "