Add Stable Diffusion 3 as library (#751)

This PR ensures downloads stats will work for https://huggingface.co/stabilityai/stable-diffusion-3-medium. --------- Co-authored-by: apolinário <[email protected]>
huggingface · Jun 21, 2024 · 6d3b5d1 · 6d3b5d1
1 parent 8207500
commit 6d3b5d1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
@@ -431,6 +431,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
  countDownloads: { term: { path: "model.safetensors" } },
  snippets: snippets.stable_audio_tools,
  },
+ "diffusion-single-file": {
+ prettyLabel: "Diffusion Single File",
+ repoName: "diffusion-single-file",
+ repoUrl: "https://github.com/comfyanonymous/ComfyUI",
+ filter: false,
+ countDownloads: {
+ wildcard: { path: "*.safetensors" },
+ },
+ },
  "stable-baselines3": {
  prettyLabel: "stable-baselines3",
  repoName: "stable-baselines3",