Nymbo commited on
Commit
4aaaa87
·
verified ·
1 Parent(s): b6a3ebe

adding the HiDream-I1 family of models

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -54,6 +54,12 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
54
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
55
  if model == 'FLUX.1 [Schnell]':
56
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 
 
 
 
 
 
57
  if model == 'Animagine 4.0':
58
  API_URL = "https://api-inference.huggingface.co/models/cagliostrolab/animagine-xl-4.0"
59
  prompt = f"masterpiece, high score, great score, absurdres, {prompt}"
@@ -456,6 +462,9 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
456
  "Flux Outfit Generator",
457
  "Frosting Lane Flux",
458
  "Half Illustration",
 
 
 
459
  "How2Draw",
460
  "Huggieverse",
461
  "Isometric 3D",
 
54
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
55
  if model == 'FLUX.1 [Schnell]':
56
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
57
+ if model == 'HiDream-I1-Full':
58
+ API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Full"
59
+ if model == 'HiDream-I1-Dev':
60
+ API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Dev"
61
+ if model == 'HiDream-I1-Fast':
62
+ API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Fast"
63
  if model == 'Animagine 4.0':
64
  API_URL = "https://api-inference.huggingface.co/models/cagliostrolab/animagine-xl-4.0"
65
  prompt = f"masterpiece, high score, great score, absurdres, {prompt}"
 
462
  "Flux Outfit Generator",
463
  "Frosting Lane Flux",
464
  "Half Illustration",
465
+ "HiDream-I1-Full",
466
+ "HiDream-I1-Dev",
467
+ "HiDream-I1-Fast",
468
  "How2Draw",
469
  "Huggieverse",
470
  "Isometric 3D",