Update src/pipeline.py

src/pipeline.py

@@ -879,8 +879,10 @@ def load_pipeline() -> Pipeline:
     inputs1 = get_example_inputs()
     print(f"AoT pre compiled path is {path}")
     
-    transformer = torch._inductor.aoti_load_package(path)
-
+    # transformer = torch._inductor.aoti_load_package(path)
+    transformer = torch._export.aot_load(path, "cuda")
+    print(f"{transformer(**inputs1)[0].shape=}")
+    
     for _ in range(2):
         _ = transformer(**inputs1)[0]