FEAT: Support sdxl-turbo (#816)

xorbitsai · Dec 26, 2023 · fa54e1b · fa54e1b
1 parent c6b133d
commit fa54e1b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/xinference/model/image/model_spec.json b/xinference/model/image/model_spec.json
@@ -5,6 +5,12 @@
     "model_id": "stabilityai/sd-turbo",
     "model_revision": "1681ed09e0cff58eeb41e878a49893228b78b94c"
   },
+  {
+    "model_name": "sdxl-turbo",
+    "model_family": "stable_diffusion",
+    "model_id": "stabilityai/sdxl-turbo",
+    "model_revision": "f4b0486b498f84668e828044de1d0c8ba486e05b"
+  },
   {
     "model_name": "stable-diffusion-v1.5",
     "model_family": "stable_diffusion",

diff --git a/xinference/model/image/tests/test_stable_diffusion.py b/xinference/model/image/tests/test_stable_diffusion.py
@@ -144,15 +144,19 @@ def test_restful_api_for_image_with_mlsd_controlnet(setup):
     logger.info("test result %s", r)
 
 
-def test_restful_api_for_sd_turbo(setup):
+@pytest.mark.parametrize("model_name", ["sd-turbo", "sdxl-turbo"])
+def test_restful_api_for_sd_turbo(setup, model_name):
+    if model_name == "sdxl-turbo":
+        pytest.skip("sdxl-turbo cost too many resources.")
+
     endpoint, _ = setup
     from ....client import Client
 
     client = Client(endpoint)
 
     model_uid = client.launch_model(
         model_uid="my_controlnet",
-        model_name="sd-turbo",
+        model_name=model_name,
         model_type="image",
     )
     model = client.get_model(model_uid)