Spaces:
Running
Running
stefanbenten
commited on
Commit
•
6dbc9d1
1
Parent(s):
f9f4daa
app.py: use samplers endpoint (#4)
Browse files- app.py: use samplers endpoint (6d9f915bd84133ee27e03ca6602afdd6737d4d8d)
app.py
CHANGED
@@ -36,6 +36,10 @@ class Prodia:
|
|
36 |
response = self._get(f"{self.base}/sdxl/models")
|
37 |
return response.json()
|
38 |
|
|
|
|
|
|
|
|
|
39 |
def _post(self, url, params):
|
40 |
headers = {
|
41 |
**self.headers,
|
@@ -119,25 +123,7 @@ with gr.Blocks(css=css) as demo:
|
|
119 |
with gr.Tab("Generation"):
|
120 |
with gr.Row():
|
121 |
with gr.Column(scale=1):
|
122 |
-
sampler = gr.Dropdown(value="DPM++ 2M Karras", show_label=True, label="Sampling Method", choices=
|
123 |
-
"Euler",
|
124 |
-
"Euler a",
|
125 |
-
"LMS",
|
126 |
-
"Heun",
|
127 |
-
"DPM2",
|
128 |
-
"DPM2 a",
|
129 |
-
"DPM++ 2S a",
|
130 |
-
"DPM++ 2M",
|
131 |
-
"DPM++ SDE",
|
132 |
-
"DPM fast",
|
133 |
-
"DPM adaptive",
|
134 |
-
"LMS Karras",
|
135 |
-
"DPM2 Karras",
|
136 |
-
"DPM2 a Karras",
|
137 |
-
"DPM++ 2S a Karras",
|
138 |
-
"DPM++ 2M Karras",
|
139 |
-
"DPM++ SDE Karras",
|
140 |
-
])
|
141 |
|
142 |
with gr.Column(scale=1):
|
143 |
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
|
|
|
36 |
response = self._get(f"{self.base}/sdxl/models")
|
37 |
return response.json()
|
38 |
|
39 |
+
def list_samplers(seflf):
|
40 |
+
response = self._get(f"{self.base}/sdxl/samplers")
|
41 |
+
return response.json()
|
42 |
+
|
43 |
def _post(self, url, params):
|
44 |
headers = {
|
45 |
**self.headers,
|
|
|
123 |
with gr.Tab("Generation"):
|
124 |
with gr.Row():
|
125 |
with gr.Column(scale=1):
|
126 |
+
sampler = gr.Dropdown(value="DPM++ 2M Karras", show_label=True, label="Sampling Method", choices=prodia_client.list_samplers())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
127 |
|
128 |
with gr.Column(scale=1):
|
129 |
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
|