Spaces:
Runtime error
Runtime error
Commit
β’
19d1ef9
1
Parent(s):
dc49fcd
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ from diffusers.utils import load_image
|
|
15 |
DESCRIPTION = """
|
16 |
# [Fluently Playground](https://huggingface.co/fluently)
|
17 |
|
|
|
18 |
"""
|
19 |
if not torch.cuda.is_available():
|
20 |
DESCRIPTION += "\n<p>Running on CPU π₯Ά This demo may not work on CPU.</p>"
|
@@ -68,7 +69,7 @@ if torch.cuda.is_available():
|
|
68 |
pipe_inpaint.to(device)
|
69 |
|
70 |
pipe_xl = StableDiffusionXLPipeline.from_pretrained(
|
71 |
-
"fluently/Fluently-XL-
|
72 |
torch_dtype=torch.float16,
|
73 |
use_safetensors=True,
|
74 |
)
|
@@ -163,7 +164,7 @@ def generate(
|
|
163 |
num_images_per_prompt=1,
|
164 |
output_type="pil",
|
165 |
).images
|
166 |
-
elif model == "Fluently XL
|
167 |
images = pipe_xl(
|
168 |
prompt=prompt,
|
169 |
negative_prompt=negative_prompt,
|
@@ -252,7 +253,7 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
|
|
252 |
with gr.Row():
|
253 |
model = gr.Radio(
|
254 |
label="Model",
|
255 |
-
choices=["Fluently XL v3 Lightning","Fluently XL
|
256 |
value="Fluently XL v3 Lightning",
|
257 |
interactive=True,
|
258 |
)
|
|
|
15 |
DESCRIPTION = """
|
16 |
# [Fluently Playground](https://huggingface.co/fluently)
|
17 |
|
18 |
+
π€ New FluentlyXL v4!
|
19 |
"""
|
20 |
if not torch.cuda.is_available():
|
21 |
DESCRIPTION += "\n<p>Running on CPU π₯Ά This demo may not work on CPU.</p>"
|
|
|
69 |
pipe_inpaint.to(device)
|
70 |
|
71 |
pipe_xl = StableDiffusionXLPipeline.from_pretrained(
|
72 |
+
"fluently/Fluently-XL-v4",
|
73 |
torch_dtype=torch.float16,
|
74 |
use_safetensors=True,
|
75 |
)
|
|
|
164 |
num_images_per_prompt=1,
|
165 |
output_type="pil",
|
166 |
).images
|
167 |
+
elif model == "Fluently XL v4":
|
168 |
images = pipe_xl(
|
169 |
prompt=prompt,
|
170 |
negative_prompt=negative_prompt,
|
|
|
253 |
with gr.Row():
|
254 |
model = gr.Radio(
|
255 |
label="Model",
|
256 |
+
choices=["Fluently XL v3 Lightning","Fluently XL v4", "Fluently v4", "Fluently Anime", "Fluently Epic", "Fluently XL v3 inpaint", "Fluently v4 inpaint"],
|
257 |
value="Fluently XL v3 Lightning",
|
258 |
interactive=True,
|
259 |
)
|