Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ if torch.cuda.is_available():
|
|
| 52 |
pipe_epic.to(device)
|
| 53 |
|
| 54 |
pipe_xl = StableDiffusionXLPipeline.from_pretrained(
|
| 55 |
-
"fluently/Fluently-XL-
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
use_safetensors=True,
|
| 58 |
)
|
|
@@ -173,7 +173,7 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
|
|
| 173 |
with gr.Row():
|
| 174 |
model = gr.Radio(
|
| 175 |
label="Model",
|
| 176 |
-
choices=["Fluently XL
|
| 177 |
value="Fluently v3.5",
|
| 178 |
interactive=True,
|
| 179 |
)
|
|
|
|
| 52 |
pipe_epic.to(device)
|
| 53 |
|
| 54 |
pipe_xl = StableDiffusionXLPipeline.from_pretrained(
|
| 55 |
+
"fluently/Fluently-XL-v2",
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
use_safetensors=True,
|
| 58 |
)
|
|
|
|
| 173 |
with gr.Row():
|
| 174 |
model = gr.Radio(
|
| 175 |
label="Model",
|
| 176 |
+
choices=["Fluently XL v2", "Fluently v3.5", "Fluently Anime", "Fluently Epic"],
|
| 177 |
value="Fluently v3.5",
|
| 178 |
interactive=True,
|
| 179 |
)
|