Skip to content

Commit

Permalink
scripts/vsmlrt.py: specify tilesize explicitly
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed Jan 21, 2022
1 parent 89d249d commit 390d0e5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/vsmlrt.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
__version__ = "3.4.1"
__version__ = "3.4.2"

__all__ = [
"Backend",
Expand Down Expand Up @@ -654,7 +654,7 @@ def inference(
clip = core.trt.Model(
clips, engine_path,
overlap=overlap,
tilesize=None if backend.static_shape else tilesize,
tilesize=tilesize,
device_id=backend.device_id,
use_cuda_graph=backend.use_cuda_graph,
num_streams=backend.num_streams,
Expand Down

0 comments on commit 390d0e5

Please sign in to comment.