fixed self.onnx_path issue in modeling_qeff

Signed-off-by: Onkar Chougule <168134249+ochougul@users.noreply.github.com>

Author: ochougul

QEfficient/base/modeling_qeff.py

@@ -295,6 +295,20 @@ def _export(
             self.onnx_path = onnx_path
         return onnx_path
 
+    def get_onnx_path(self, prefill_only: Optional[bool] = False,
+                      specializations: Optional[List[Dict[str, int]]] = None, 
+                      offload_pt_weights: Optional[bool] = True):
+        kwargs = {"offload_pt_weights": offload_pt_weights}
+        if prefill_only:
+            if self.prefill_onnx_path is None:
+                kwargs.update({"prefill_only": prefill_only, "prefill_seq_len": specializations[0].get("seq_len")})
+                self.export(**kwargs)
+            return self.prefill_onnx_path
+        else:
+            if self.onnx_path is None:
+                self.export(**kwargs)
+            return self.onnx_path
+
     @dump_qconfig
     def _compile(
         self,
@@ -335,17 +349,7 @@ def _compile(
 
                 For QNN Compilation path, when enable_qnn is set to True, any parameter passed in compiler_options will be ignored.
         """
-        kwargs = {"offload_pt_weights": offload_pt_weights}
-        if onnx_path is None and prefill_only:
-            kwargs.update({"prefill_only": prefill_only, "prefill_seq_len": specializations[0].get("seq_len")})
-            self.export(**kwargs)
-            onnx_path = Path(self.prefill_onnx_path)
-        elif onnx_path is None:
-            self.export(**kwargs)
-            onnx_path = Path(self.onnx_path)
-        else:
-            onnx_path = Path(onnx_path)
-
+        onnx_path = Path(onnx_path if onnx_path else self.get_onnx_path(prefill_only, specializations, offload_pt_weights))
         compile_dir = Path(compile_dir or onnx_path.parent)
         qpc_path = compile_dir / "qpc"
         if not onnx_path.is_file():