Skip to content

Commit bbec0b5

Browse files
mark as run slow to test with CI
1 parent f8b54c6 commit bbec0b5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/openvino/test_diffusion.py

+4
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import unittest
1616

1717
import numpy as np
18+
import pytest
1819
import torch
1920
from diffusers import (
2021
AutoPipelineForImage2Image,
@@ -296,6 +297,7 @@ def test_height_width_properties(self, model_arch: str):
296297
self.assertEqual(ov_pipeline.height, height)
297298
self.assertEqual(ov_pipeline.width, width)
298299

300+
@pytest.mark.run_slow
299301
@slow
300302
@require_diffusers
301303
def test_textual_inversion(self):
@@ -507,6 +509,7 @@ def test_height_width_properties(self, model_arch: str):
507509
self.assertEqual(ov_pipeline.height, height)
508510
self.assertEqual(ov_pipeline.width, width)
509511

512+
@pytest.mark.run_slow
510513
@slow
511514
@require_diffusers
512515
def test_textual_inversion(self):
@@ -723,6 +726,7 @@ def test_height_width_properties(self, model_arch: str):
723726
self.assertEqual(ov_pipeline.height, height)
724727
self.assertEqual(ov_pipeline.width, width)
725728

729+
@pytest.mark.run_slow
726730
@slow
727731
@require_diffusers
728732
def test_textual_inversion(self):

0 commit comments

Comments
 (0)