Update optimum/onnxruntime/modeling_diffusion.py

Co-authored-by: Ella Charlaix <[email protected]>
huggingface · Oct 3, 2024 · 2e242e5 · 2e242e5
1 parent 15c94bc
commit 2e242e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)