dippoo Claude Opus 4.5 commited on
Commit
4b9b1b8
·
1 Parent(s): 4deeec0

Add Qwen and NSFW-friendly models

Browse files

Text-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="NanoBanana (Google)">
887
- <option value="nano-banana-pro">NanoBanana Pro</option>
888
- <option value="nano-banana">NanoBanana</option>
 
 
 
 
 
 
 
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="qwen-image">Qwen Image</option>
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 (WaveSpeed optimized)
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
  }