From df8c62618d00ad879fe0b2526f33b2d5e274d1e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E4=BD=B3=E5=AE=9D?= Date: Fri, 16 Aug 2024 16:38:31 +0800 Subject: [PATCH] Update models.py: maybe typo maybe it should be rename output[0]? sorry i don't understand the code very well, but remove feels kinda strange.. --- demo/Diffusion/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/Diffusion/models.py b/demo/Diffusion/models.py index d9d82b69..06929dad 100644 --- a/demo/Diffusion/models.py +++ b/demo/Diffusion/models.py @@ -596,7 +596,7 @@ def optimize(self, onnx_graph): opt.infer_shapes() opt.info(self.name + ': shape inference') opt.select_outputs([0], names=['text_embeddings']) # rename network output - opt.info(self.name + ': remove output[0]') + opt.info(self.name + ': rename output[0]') opt_onnx_graph = opt.cleanup(return_onnx=True) if 'hidden_states' in self.extra_output_names: opt_onnx_graph = opt.clip_add_hidden_states(return_onnx=True)