Skip to content

Commit 76f01fc

Browse files
Fix import sorting
Signed-off-by: Anish Shanbhag <ashanbhag@nvidia.com>
1 parent 321f28f commit 76f01fc

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

tensorrt_llm/llmapi/llm_args.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
from pydantic import Field as PydanticField
1818
from pydantic import PrivateAttr, field_validator, model_validator
1919
from strenum import StrEnum
20-
2120
from transformers import PreTrainedTokenizerBase
2221

2322
try:

tensorrt_llm/llmapi/llm_utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,10 @@
99
from typing import Any, Callable, List, Optional, Tuple, Union
1010

1111
import torch
12+
import transformers
1213
from pydantic import BaseModel
1314
from tqdm import tqdm
1415

15-
import transformers
16-
1716
from .._utils import (global_mpi_rank, local_mpi_rank, mpi_barrier,
1817
mpi_broadcast, mpi_rank, release_gc)
1918
# yapf: disable

0 commit comments

Comments
 (0)