Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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",
|