Spaces:
Running
Running
adding the HiDream-I1 family of models
Browse files
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",
|