From 2e242e5d6da2339e3ef63ceaec31a670619566e1 Mon Sep 17 00:00:00 2001 From: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> Date: Thu, 3 Oct 2024 17:15:04 +0200 Subject: [PATCH] Update optimum/onnxruntime/modeling_diffusion.py Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> --- optimum/onnxruntime/modeling_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py index 14e41f27818..bf6dc002f9e 100644 --- a/optimum/onnxruntime/modeling_diffusion.py +++ b/optimum/onnxruntime/modeling_diffusion.py @@ -560,7 +560,7 @@ def prepare_onnx_outputs( @abstractmethod def forward(self, *args, **kwargs): - pass + raise NotImplementedError def __call__(self, *args, **kwargs): return self.forward(*args, **kwargs)