changed the file place and deleted unnecessary code

Author: cehongwang

demo/flux_demo.py examples/apps/flux_demo.py

@@ -88,6 +88,7 @@
 del ep
 pipe.transformer = trt_gm
 pipe.transformer.config = config
+trt_gm.device = torch.device("cuda")
 torch.cuda.empty_cache()
 
 

py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py

@@ -706,10 +706,9 @@ def run(
             builder_config, self.compilation_settings.timing_cache_path
         )
         # TODO: Memory control prototyping. Under discussion
-        self.module.to("cpu")
-        torch.cuda.empty_cache()
         del self.module
         gc.collect()
+        torch.cuda.empty_cache()
         serialized_engine = self.builder.build_serialized_network(
             self.ctx.net, builder_config
         )