From 013c6e7fc3bdeb0680674789b2ed164825b3e8ee Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Wed, 7 Dec 2022 18:45:49 +0800
Subject: [PATCH] update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 clip_vision.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clip_vision.py b/clip_vision.py
index dd341df..8c18f60 100644
--- a/clip_vision.py
+++ b/clip_vision.py
@@ -68,7 +68,7 @@ class ClipVision(NNOperator):
             return False
         x = torch.randn((1, 3, 224, 224))
 
-        torch.onnx.export(self.model, x, output_file, input_names=['INPUT0'],
+        torch.onnx.export(self.model.model, x, output_file, input_names=['INPUT0'],
                           output_names=['OUTPUT0'],  dynamic_axes={'INPUT0': [0]})
         return True