From e83bb70324696029f687ff0338fd71314647bffe Mon Sep 17 00:00:00 2001
From: Nirant <NirantK@users.noreply.github.com>
Date: Mon, 4 Mar 2024 14:37:58 +0530
Subject: [PATCH] Update experiments/attention_export.py

Co-authored-by: Anush <anushshetty90@gmail.com>
---
 experiments/attention_export.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/experiments/attention_export.py b/experiments/attention_export.py
index aff6fbf9..36a10698 100644
--- a/experiments/attention_export.py
+++ b/experiments/attention_export.py
@@ -11,7 +11,7 @@
 
 # export if the output model does not exist
 # try:
-#     sess = onnxruntime.InferenceSession(f"{output_dir}/{model_id.replace('/', '_')}.onnx")
+#     sess = onnxruntime.InferenceSession(f"{output_dir}/model.onnx")
 #     print("Model already exported")
 # except FileNotFoundError:
 print(f"Exporting model to {output_dir}")