Skip to content

Commit 25862ad

Browse files
committed
Fix style
1 parent 11b43ca commit 25862ad

File tree

3 files changed

+3
-5
lines changed

3 files changed

+3
-5
lines changed

all_models/gpt/preprocessing/1/model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,10 @@
55
import numpy as np
66
import torch
77
import triton_python_backend_utils as pb_utils
8+
from tensorrt_llm.runtime import to_word_list_format
89
from torch.nn.utils.rnn import pad_sequence
910
from transformers import AutoTokenizer, LlamaTokenizer, T5Tokenizer
1011

11-
from tensorrt_llm.runtime import to_word_list_format
12-
1312

1413
class TritonPythonModel:
1514
"""Your Python model must use the same class name. Every Python model

all_models/gpt/tensorrt_llm/1/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33

44
import torch
55
import triton_python_backend_utils as pb_utils
6+
from tensorrt_llm.runtime import GenerationSession, ModelConfig, SamplingConfig
67
from torch import from_numpy
78

89
import tensorrt_llm
9-
from tensorrt_llm.runtime import GenerationSession, ModelConfig, SamplingConfig
1010

1111

1212
def mpi_comm():

all_models/inflight_batcher_llm/preprocessing/1/model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,10 @@
55
import numpy as np
66
import torch
77
import triton_python_backend_utils as pb_utils
8+
from tensorrt_llm.runtime import to_word_list_format
89
from torch.nn.utils.rnn import pad_sequence
910
from transformers import AutoTokenizer, LlamaTokenizer, T5Tokenizer
1011

11-
from tensorrt_llm.runtime import to_word_list_format
12-
1312

1413
class TritonPythonModel:
1514
"""Your Python model must use the same class name. Every Python model

0 commit comments

Comments
 (0)