TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Results 599 TensorRT issues
Sort by recently updated
recently updated
newest added

Load checkpoint from: wav2lip_checkpoint/checkpoint_step002680000.pth 0%| | 0/22 [01:37

bug

Hi, I have trained a model whose input is a set of 3D points with a shape `Nx3`, N is not a fixed number. In this case, how to set...

question

- Rename to `enforce_types` - Allow `(int, float, bool)` type requirements - Allow typing of tensor data to streamline casting

- Pretty-Print Compilation Settings - Add Torch subgraph blocks for fast partitioner at least See https://github.com/pytorch/TensorRT/pull/2451#pullrequestreview-1787640029 See also: https://github.com/pytorch/TensorRT/pull/2451#discussion_r1433077927

Port #2445 to apply to TorchScript path. Requires enablement of an additional (optional) compile-time argument in TorchScript and builder flag in TensorRT

feature request

## ❓ Question Hello, I have a torch scripted model that I am trying to compile with TensorRT: ```py import cv2 import numpy as np import torch from torchvision.transforms import...

question
component: lowering

Bumps [transformers](https://github.com/huggingface/transformers) from 4.33.2 to 4.36.0. Release notes Sourced from transformers's releases. v4.36: Mixtral, Llava/BakLlava, SeamlessM4T v2, AMD ROCm, F.sdpa wide-spread support New model additions Mixtral Mixtral is the new...

component: dependencies
cla signed
python

## ❓ Question Hello. I am currently trying to infer the stable diffusion XL inpaint model using your package. model link : https://huggingface.co/diffusers/stable-diffusion-xl-1.0-inpainting-0.1 I referred to your example code and...

question

# Description Support 1d ITensor offsets for embedding_bag converter. Note that this is only for 1d inputs. There's a bug that same offsets with different types (tensor or ITensor) when...

component: tests
component: lowering
component: conversion
component: converters
component: api [Python]
cla signed
component: dynamo