dru10 comments

Repositories
Issues
Comments

Results 1 comments of


                                            dru10

❓ [Question] How do you save a unet model compiled Torch-TensorRT (Stable Diffusion XL)

> After altering the code from torch export (which probably wasn't necessary) For reference, this is the modification I did inside [`py/torch_tensorrt/dynamo/_tracer.py#L81`](https://github.com/pytorch/TensorRT/blob/622ca532fcc282fc14b9b7badfd24d43c593266c/py/torch_tensorrt/dynamo/_tracer.py#L81) `exp_program = export(mod, tuple(torch_inputs), kwargs=kwargs, dynamic_shapes=tuple(dynamic_shapes))` And this...