fix(api): fully qualify HF models (#371)

ssube · May 5, 2023 · da6aa15 · da6aa15
1 parent 88fa126
commit da6aa15
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/api/extras.json b/api/extras.json
@@ -2,11 +2,11 @@
   "diffusion": [
     {
       "name": "diffusion-knollingcase",
-      "source": "Aybeeceedee/knollingcase"
+      "source": "huggingface://Aybeeceedee/knollingcase"
     },
     {
       "name": "diffusion-openjourney",
-      "source": "prompthero/openjourney"
+      "source": "huggingface://prompthero/openjourney"
     },
     {
       "name": "diffusion-stablydiffused-aesthetic-v2-6",
@@ -20,27 +20,27 @@
     },
     {
       "name": "stable-diffusion-onnx-v1-5",
-      "source": "runwayml/stable-diffusion-v1-5",
+      "source": "huggingface://runwayml/stable-diffusion-v1-5",
       "inversions": [
         {
           "name": "line-art",
-          "source": "sd-concepts-library/line-art"
+          "source": "huggingface://sd-concepts-library/line-art"
         },
         {
           "name": "cubex",
-          "source": "sd-concepts-library/cubex"
+          "source": "huggingface://sd-concepts-library/cubex"
         },
         {
           "name": "birb",
-          "source": "sd-concepts-library/birb-style"
+          "source": "huggingface://sd-concepts-library/birb-style"
         },
         {
           "name": "minecraft",
-          "source": "sd-concepts-library/minecraft-concept-art"
+          "source": "huggingface://sd-concepts-library/minecraft-concept-art"
         },
         {
           "name": "ugly-sonic",
-          "source": "sd-concepts-library/ugly-sonic"
+          "source": "huggingface://sd-concepts-library/ugly-sonic"
         }
       ]
     }

diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py
@@ -65,7 +65,6 @@
             "stable-diffusion-onnx-v1-inpainting",
             model_source_huggingface + "runwayml/stable-diffusion-inpainting",
         ),
-        # TODO: should have its own converter
         (
             "upscaling-stable-diffusion-x4",
             model_source_huggingface + "stabilityai/stable-diffusion-x4-upscaler",