feat: Update trending models

huuquyet · Aug 30, 2024 · 8aff43c · 8aff43c
1 parent cb8c46b
commit 8aff43c
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 28 deletions.
diff --git a/README.md b/README.md
@@ -9,8 +9,8 @@ This action generate awesome wallpapers using HuggingFace Inference API (serverl
 <div align="center">
   <img alt="Awesome Wallpapers" src="./assets/wallpaper.jpg">
 
-<!-- START_CAPTION -->
-*A pristine, coral reef teeming with colorful tropical fish beneath the azure ocean surface* by [stabilityai/stable-diffusion-2-1](https://hf.co/stabilityai/stable-diffusion-2-1)
+<!-- START_CAPTION -->
+*A pristine, coral reef teeming with colorful tropical fish beneath the azure ocean surface* by [stabilityai/stable-diffusion-2-1](https://hf.co/stabilityai/stable-diffusion-2-1)
 <!-- END_CAPTION -->
 </div>
 
@@ -27,12 +27,12 @@ Then add to the [secrets for the repository](https://docs.github.com/en/actions/
 
 ## List of models
 
-- [runwayml/stable-diffusion-v1-5](https://hf.co/runwayml/stable-diffusion-v1-5)
-- [CompVis/stable-diffusion-v1-4](https://hf.co/CompVis/stable-diffusion-v1-4)
-- [stabilityai/stable-diffusion-xl-base-1.0](https://hf.co/stabilityai/stable-diffusion-xl-base-1.0)
-- [stabilityai/stable-diffusion-2-1](https://hf.co/stabilityai/stable-diffusion-2-1)
-- [prompthero/openjourney](https://hf.co/prompthero/openjourney)
-- [prompthero/openjourney-v4](https://hf.co/prompthero/openjourney-v4)
+- [black-forest-labs/FLUX.1-dev](https://hf.co/black-forest-labs/FLUX.1-dev)
+- [black-forest-labs/FLUX.1-schnell](https://hf.co/black-forest-labs/FLUX.1-schnell)
+- [ByteDance/Hyper-SD](https://hf.co/ByteDance/Hyper-SD)
+- [latent-consistency/lcm-lora-sdxl](https://hf.co/latent-consistency/lcm-lora-sdxl)
+- [Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro](https://hf.co/Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro)
+- [stabilityai/stable-diffusion-3-medium-diffusers](https://hf.co/stabilityai/stable-diffusion-3-medium-diffusers)
 
 ## Outputs
 

diff --git a/bun.lockb b/bun.lockb
diff --git a/dist/index.js b/dist/index.js
@@ -26839,12 +26839,12 @@ const prompts_namespaceObject = JSON.parse('["A pristine, turquoise glacier lake
 
 // The list of text-to-image models that support inference API
 const MODELS = [
-    'runwayml/stable-diffusion-v1-5',
-    'CompVis/stable-diffusion-v1-4',
-    'stabilityai/stable-diffusion-xl-base-1.0',
-    'stabilityai/stable-diffusion-2-1',
-    'prompthero/openjourney',
-    'prompthero/openjourney-v4',
+    'black-forest-labs/FLUX.1-dev',
+    'black-forest-labs/FLUX.1-schnell',
+    'ByteDance/Hyper-SD',
+    'latent-consistency/lcm-lora-sdxl',
+    'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro',
+    'stabilityai/stable-diffusion-3-medium-diffusers',
 ];
 // The patterns to set the caption of image
 const START_CAPTION = '<!-- START_CAPTION -->';

diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/package.json b/package.json
@@ -25,7 +25,7 @@
     "@actions/core": "^1.10.1"
   },
   "devDependencies": {
-    "@types/bun": "^1.1.6",
+    "@types/bun": "^1.1.8",
     "@vercel/ncc": "^0.38.1",
     "typescript": "^5.5.4"
   },

diff --git a/src/generate.py b/src/generate.py
@@ -7,12 +7,12 @@
 API_TOKEN = getenv("HF_API_TOKEN")
 # The list of text-to-image models that support inference API
 MODELS = [
-    "runwayml/stable-diffusion-v1-5",
-    "CompVis/stable-diffusion-v1-4",
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    "stabilityai/stable-diffusion-2-1",
-    "prompthero/openjourney",
-    "prompthero/openjourney-v4",
+    'black-forest-labs/FLUX.1-dev',
+    'black-forest-labs/FLUX.1-schnell',
+    'ByteDance/Hyper-SD',
+    'latent-consistency/lcm-lora-sdxl',
+    'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro',
+    'stabilityai/stable-diffusion-3-medium-diffusers',
 ]
 
 random_model = random.choice(MODELS)

diff --git a/src/utils.ts b/src/utils.ts
@@ -4,12 +4,12 @@ import { default as prompts } from '../assets/prompts.json'
 
 // The list of text-to-image models that support inference API
 const MODELS: string[] = [
-  'runwayml/stable-diffusion-v1-5',
-  'CompVis/stable-diffusion-v1-4',
-  'stabilityai/stable-diffusion-xl-base-1.0',
-  'stabilityai/stable-diffusion-2-1',
-  'prompthero/openjourney',
-  'prompthero/openjourney-v4',
+  'black-forest-labs/FLUX.1-dev',
+  'black-forest-labs/FLUX.1-schnell',
+  'ByteDance/Hyper-SD',
+  'latent-consistency/lcm-lora-sdxl',
+  'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro',
+  'stabilityai/stable-diffusion-3-medium-diffusers',
 ]
 
 // The patterns to set the caption of image