Correct the flux export example

Author: cehongwang

examples/dynamo/torch_export_flux_dev.py

@@ -113,6 +113,7 @@
     use_fp32_acc=True,
     use_explicit_typing=True,
     use_python_runtime=True,
+    immutable_weights=False,
 )
 
 # %%
@@ -121,11 +122,11 @@
 # Release the GPU memory occupied by the exported program and the pipe.transformer
 # Set the transformer in the Flux pipeline to the Torch-TRT compiled model
 
-del ep
-backbone.to("cpu")
 pipe.to(DEVICE)
-torch.cuda.empty_cache()
+backbone.to("cpu")
 pipe.transformer = trt_gm
+del ep
+torch.cuda.empty_cache()
 pipe.transformer.config = config
 trt_gm.device = torch.device("cuda")
 # %%