pytorch-lightning

39e24f5d - Torch-Tensorrt Integration with LightningModule (#20808)

Torch-Tensorrt Integration with LightningModule (#20808) * feat: add `to_tensorrt` in the `LightningModule`. * refactor: fix `to_tensorrt` impl * test: add test_torch_tensorrt.py * add dependency in test requirement * limit the torch-tensorrt condition again * update tensorrt version * update tensorrt source * update test.txt * ci: add extra-index * ci: use find-links instead. works on my computer... * fix: fix bug in torch-tensorrt 2.8.0 * add find links in ci test. * chlog --------- Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Jirka B <j.borovec+github@gmail.com>