Update experiments/attention_export.py

Co-authored-by: Anush <[email protected]>
qdrant · Mar 4, 2024 · e83bb70 · e83bb70
1 parent 96d0099
commit e83bb70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}")