Skip to content

Commit

Permalink
fix isort error
Browse files Browse the repository at this point in the history
  • Loading branch information
bmullick-amd authored Dec 18, 2024
1 parent b554266 commit 2a8bb2e
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions vllm/model_executor/models/t5.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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 "
Expand Down

0 comments on commit 2a8bb2e

Please sign in to comment.