Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,13 +19,13 @@ from meta import (
|
|
19 |
)
|
20 |
|
21 |
|
22 |
-
sdxl_pipe = load_model(model_id="KBlueLeaf/Kohaku-XL-
|
23 |
models.load_model(models.model_list[0])
|
24 |
models.text_model.cpu()
|
25 |
torch.cuda.empty_cache()
|
26 |
|
27 |
current_dtg_model = models.model_list[0]
|
28 |
-
current_sdxl_model = "KBlueLeaf/Kohaku-XL-
|
29 |
|
30 |
model_loading_lock = Lock()
|
31 |
model_running_lock = Lock()
|
@@ -171,7 +171,7 @@ click "Next" button until you get the dragon girl you like.
|
|
171 |
style = gr.Dropdown(
|
172 |
DEFAULT_STYLE_LIST,
|
173 |
label="Style",
|
174 |
-
value=list(DEFAULT_STYLE_LIST)[
|
175 |
)
|
176 |
submit = gr.Button("Next", variant="primary")
|
177 |
dtg_output = gr.TextArea(
|
|
|
19 |
)
|
20 |
|
21 |
|
22 |
+
sdxl_pipe = load_model(model_id="KBlueLeaf/Kohaku-XL-Zeta", device="cuda")
|
23 |
models.load_model(models.model_list[0])
|
24 |
models.text_model.cpu()
|
25 |
torch.cuda.empty_cache()
|
26 |
|
27 |
current_dtg_model = models.model_list[0]
|
28 |
+
current_sdxl_model = "KBlueLeaf/Kohaku-XL-Zeta"
|
29 |
|
30 |
model_loading_lock = Lock()
|
31 |
model_running_lock = Lock()
|
|
|
171 |
style = gr.Dropdown(
|
172 |
DEFAULT_STYLE_LIST,
|
173 |
label="Style",
|
174 |
+
value=list(DEFAULT_STYLE_LIST)[-2],
|
175 |
)
|
176 |
submit = gr.Button("Next", variant="primary")
|
177 |
dtg_output = gr.TextArea(
|