Spaces:
Running
Running
Add Qwen and NSFW-friendly models
Browse filesText-to-Image:
- Qwen 2.5 VL 72B, Qwen 2 VL 72B (less restrictive)
- Pony Diffusion v6/v7 (anime/NSFW friendly)
- Illustrious XL (anime)
- FLUX Dev Uncensored
- SDXL Lightning, SDXL Turbo
Organized dropdown with "NSFW Friendly" section for easy access
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
src/content_engine/api/ui.html
CHANGED
|
@@ -878,27 +878,42 @@ select { cursor: pointer; }
|
|
| 878 |
<div id="cloud-model-select" style="display:none">
|
| 879 |
<label>Cloud Model (Text-to-Image)</label>
|
| 880 |
<select id="gen-cloud-model">
|
| 881 |
-
<optgroup label="SeeDream (ByteDance)">
|
| 882 |
<option value="seedream-4.5">SeeDream v4.5 (Best Quality)</option>
|
| 883 |
<option value="seedream-4">SeeDream v4</option>
|
| 884 |
<option value="seedream-3.1">SeeDream v3.1</option>
|
| 885 |
</optgroup>
|
| 886 |
-
<optgroup label="
|
| 887 |
-
<option value="
|
| 888 |
-
<option value="
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 889 |
</optgroup>
|
| 890 |
<optgroup label="FLUX (Black Forest Labs)">
|
| 891 |
<option value="flux-pro">FLUX Pro (Highest Quality)</option>
|
| 892 |
<option value="flux-dev">FLUX Dev</option>
|
| 893 |
<option value="flux-schnell">FLUX Schnell (Fast)</option>
|
| 894 |
</optgroup>
|
|
|
|
|
|
|
|
|
|
|
|
|
| 895 |
<optgroup label="WAN (Alibaba)">
|
| 896 |
<option value="wan-2.6">WAN 2.6 (Latest)</option>
|
| 897 |
<option value="wan-2.2">WAN 2.2</option>
|
| 898 |
</optgroup>
|
|
|
|
|
|
|
|
|
|
|
|
|
| 899 |
<optgroup label="Other">
|
| 900 |
<option value="dreamina-3.1">Dreamina v3.1</option>
|
| 901 |
-
<option value="
|
| 902 |
</optgroup>
|
| 903 |
</select>
|
| 904 |
</div>
|
|
|
|
| 878 |
<div id="cloud-model-select" style="display:none">
|
| 879 |
<label>Cloud Model (Text-to-Image)</label>
|
| 880 |
<select id="gen-cloud-model">
|
| 881 |
+
<optgroup label="SeeDream (ByteDance) - NSFW OK">
|
| 882 |
<option value="seedream-4.5">SeeDream v4.5 (Best Quality)</option>
|
| 883 |
<option value="seedream-4">SeeDream v4</option>
|
| 884 |
<option value="seedream-3.1">SeeDream v3.1</option>
|
| 885 |
</optgroup>
|
| 886 |
+
<optgroup label="Qwen (Alibaba) - Less Restrictive">
|
| 887 |
+
<option value="qwen2.5-vl">Qwen 2.5 VL 72B (Best)</option>
|
| 888 |
+
<option value="qwen2-vl">Qwen 2 VL 72B</option>
|
| 889 |
+
<option value="qwen-image">Qwen Image</option>
|
| 890 |
+
</optgroup>
|
| 891 |
+
<optgroup label="NSFW Friendly">
|
| 892 |
+
<option value="pony-v7">Pony Diffusion v7 (Anime/NSFW)</option>
|
| 893 |
+
<option value="pony-v6">Pony Diffusion v6</option>
|
| 894 |
+
<option value="illustrious-xl">Illustrious XL (Anime)</option>
|
| 895 |
+
<option value="flux-dev-uncensored">FLUX Dev Uncensored</option>
|
| 896 |
</optgroup>
|
| 897 |
<optgroup label="FLUX (Black Forest Labs)">
|
| 898 |
<option value="flux-pro">FLUX Pro (Highest Quality)</option>
|
| 899 |
<option value="flux-dev">FLUX Dev</option>
|
| 900 |
<option value="flux-schnell">FLUX Schnell (Fast)</option>
|
| 901 |
</optgroup>
|
| 902 |
+
<optgroup label="NanoBanana (Google)">
|
| 903 |
+
<option value="nano-banana-pro">NanoBanana Pro</option>
|
| 904 |
+
<option value="nano-banana">NanoBanana</option>
|
| 905 |
+
</optgroup>
|
| 906 |
<optgroup label="WAN (Alibaba)">
|
| 907 |
<option value="wan-2.6">WAN 2.6 (Latest)</option>
|
| 908 |
<option value="wan-2.2">WAN 2.2</option>
|
| 909 |
</optgroup>
|
| 910 |
+
<optgroup label="SDXL Variants">
|
| 911 |
+
<option value="sdxl-lightning">SDXL Lightning (Fast)</option>
|
| 912 |
+
<option value="sdxl-turbo">SDXL Turbo</option>
|
| 913 |
+
</optgroup>
|
| 914 |
<optgroup label="Other">
|
| 915 |
<option value="dreamina-3.1">Dreamina v3.1</option>
|
| 916 |
+
<option value="dreamina-3">Dreamina v3</option>
|
| 917 |
</optgroup>
|
| 918 |
</select>
|
| 919 |
</div>
|
src/content_engine/services/cloud_providers/wavespeed_provider.py
CHANGED
|
@@ -38,7 +38,7 @@ MODEL_MAP = {
|
|
| 38 |
# NanoBanana (Google)
|
| 39 |
"nano-banana": "google/nano-banana",
|
| 40 |
"nano-banana-pro": "google/nano-banana-pro",
|
| 41 |
-
# SeeDream (ByteDance)
|
| 42 |
"seedream-3": "bytedance/seedream-v3",
|
| 43 |
"seedream-3.1": "bytedance/seedream-v3.1",
|
| 44 |
"seedream-4": "bytedance/seedream-v4",
|
|
@@ -48,13 +48,24 @@ MODEL_MAP = {
|
|
| 48 |
"wan-2.6": "alibaba/wan-2.6",
|
| 49 |
# FLUX (Black Forest Labs)
|
| 50 |
"flux-dev": "black-forest-labs/flux-dev",
|
|
|
|
| 51 |
"flux-schnell": "black-forest-labs/flux-schnell",
|
| 52 |
"flux-pro": "black-forest-labs/flux-pro",
|
| 53 |
# Dreamina (ByteDance)
|
| 54 |
"dreamina-3": "bytedance/dreamina-v3",
|
| 55 |
"dreamina-3.1": "bytedance/dreamina-v3.1",
|
| 56 |
-
# Qwen (
|
| 57 |
"qwen-image": "wavespeedai/qwen-image",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
# Default
|
| 59 |
"default": "bytedance/seedream-v4.5",
|
| 60 |
}
|
|
|
|
| 38 |
# NanoBanana (Google)
|
| 39 |
"nano-banana": "google/nano-banana",
|
| 40 |
"nano-banana-pro": "google/nano-banana-pro",
|
| 41 |
+
# SeeDream (ByteDance) - good for NSFW
|
| 42 |
"seedream-3": "bytedance/seedream-v3",
|
| 43 |
"seedream-3.1": "bytedance/seedream-v3.1",
|
| 44 |
"seedream-4": "bytedance/seedream-v4",
|
|
|
|
| 48 |
"wan-2.6": "alibaba/wan-2.6",
|
| 49 |
# FLUX (Black Forest Labs)
|
| 50 |
"flux-dev": "black-forest-labs/flux-dev",
|
| 51 |
+
"flux-dev-uncensored": "black-forest-labs/flux-dev-uncensored",
|
| 52 |
"flux-schnell": "black-forest-labs/flux-schnell",
|
| 53 |
"flux-pro": "black-forest-labs/flux-pro",
|
| 54 |
# Dreamina (ByteDance)
|
| 55 |
"dreamina-3": "bytedance/dreamina-v3",
|
| 56 |
"dreamina-3.1": "bytedance/dreamina-v3.1",
|
| 57 |
+
# Qwen (Alibaba) - less restrictive
|
| 58 |
"qwen-image": "wavespeedai/qwen-image",
|
| 59 |
+
"qwen2-vl": "alibaba/qwen2-vl-72b",
|
| 60 |
+
"qwen2.5-vl": "alibaba/qwen2.5-vl-72b",
|
| 61 |
+
# Illustrious XL - anime/NSFW friendly
|
| 62 |
+
"illustrious-xl": "illustrious/illustrious-xl-v1",
|
| 63 |
+
# Pony Diffusion - NSFW friendly
|
| 64 |
+
"pony-v6": "pony-diffusion/pony-v6",
|
| 65 |
+
"pony-v7": "pony-diffusion/pony-v7",
|
| 66 |
+
# SDXL variants
|
| 67 |
+
"sdxl-turbo": "stabilityai/sdxl-turbo",
|
| 68 |
+
"sdxl-lightning": "bytedance/sdxl-lightning",
|
| 69 |
# Default
|
| 70 |
"default": "bytedance/seedream-v4.5",
|
| 71 |
}
|