fix rife trt

Former-commit-id: 712631c
Former-commit-id: e27f23d

backend/src/TensorRTHandler.py

@@ -68,7 +68,6 @@ def export_dynamo_model(
     ):
         """Exports a model using TensorRT Dynamo."""
         model.to(device=device, dtype=dtype)
-        example_inputs = [input.to(device=device,dtype=dtype) for input in example_inputs]
         exported_program = torch.export.export(
             model, tuple(example_inputs), dynamic_shapes=None
         )
@@ -108,8 +107,7 @@ def export_torchscript_model(
 
         # maybe try to load it onto CUDA, and clear pytorch cache after.
         model.to(device=device,dtype=dtype)
-        example_inputs = [input.to(device=device,dtype=dtype) for input in example_inputs]
-        module = torch.jit.trace(model, example_inputs) # have to put both on same device or sum
+        module = torch.jit.trace(model, example_inputs) 
         torch.cuda.empty_cache()
         model = None