sdxl support cpu_text_encoder

2024-01-10 13:34:11 +08:00
parent 05a15b2e1f
commit 38b6edacf0
5 changed files with 80 additions and 9 deletions
--- a/iopaint/tests/utils.py
+++ b/iopaint/tests/utils.py
@@ -17,7 +17,7 @@ def check_device(device: str) -> int:
        pytest.skip("CUDA is not available, skip test on cuda")
    if device == "mps" and not torch.backends.mps.is_available():
        pytest.skip("mps is not available, skip test on mps")
-    steps = 1 if device == "cpu" else 20
+    steps = 2 if device == "cpu" else 20
    return steps