Skip to content

Commit 49e403f

Browse files
registry file import ordering
Signed-off-by: yechank <[email protected]>
1 parent 2af0372 commit 49e403f

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tensorrt_llm/inputs/registry.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,15 @@
11
import enum
22
import random
3+
from abc import ABC, abstractmethod
34
from dataclasses import dataclass, field
45
from typing import (Any, Callable, Dict, List, Optional, Protocol, Tuple, Type,
56
TypeVar)
67

78
import torch
89
from PIL import Image
910
from torch import Tensor, nn
11+
from transformers import (AutoProcessor, PretrainedConfig,
12+
PreTrainedTokenizerBase)
1013

1114
import tensorrt_llm
1215

@@ -22,11 +25,6 @@
2225

2326
ExtraProcessedInputs = Dict[str, Any]
2427

25-
from abc import ABC, abstractmethod
26-
27-
from transformers import (AutoProcessor, PretrainedConfig,
28-
PreTrainedTokenizerBase)
29-
3028

3129
class InputProcessor(Protocol):
3230
"""

0 commit comments

Comments
 (0)