diff --git a/lama_cleaner/tests/test_paint_by_example.py b/lama_cleaner/tests/test_paint_by_example.py
index a8ce09a..476664f 100644
--- a/lama_cleaner/tests/test_paint_by_example.py
+++ b/lama_cleaner/tests/test_paint_by_example.py
@@ -38,7 +38,7 @@ def assert_equal(
 @pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
 def test_paint_by_example(strategy):
     model = ModelManager(name="paint_by_example", device=device)
-    cfg = get_config(strategy, paint_by_example_steps=30)
+    cfg = get_config(strategy, paint_by_example_steps=30 if device == 'cuda' else 1)
     assert_equal(
         model,
         cfg,
diff --git a/lama_cleaner/tests/test_sd_model.py b/lama_cleaner/tests/test_sd_model.py
index 6abc261..eb573be 100644
--- a/lama_cleaner/tests/test_sd_model.py
+++ b/lama_cleaner/tests/test_sd_model.py
@@ -1,12 +1,10 @@
-import os
 from pathlib import Path
 
-import cv2
 import pytest
 import torch
 
 from lama_cleaner.model_manager import ModelManager
-from lama_cleaner.schema import Config, HDStrategy, LDMSampler, SDSampler
+from lama_cleaner.schema import HDStrategy, SDSampler
 from lama_cleaner.tests.test_model import get_config, assert_equal
 
 current_dir = Path(__file__).parent.absolute().resolve()
@@ -96,7 +94,7 @@ def test_runway_sd_1_5_negative_prompt(sd_device, strategy, sampler):
     if sd_device == 'cuda' and not torch.cuda.is_available():
         return
 
-    sd_steps = 50
+    sd_steps = 50 if sd_device == 'cuda' else 1
     model = ModelManager(name="sd1.5",
                          device=torch.device(sd_device),
                          hf_access_token="",