Spaces:
Runtime error
Runtime error
refiner setup
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0")
|
|
32 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
|
33 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
|
34 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
35 |
-
ENABLE_REFINER = os.getenv("ENABLE_REFINER", "0")
|
36 |
|
37 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
38 |
if torch.cuda.is_available():
|
@@ -219,14 +219,14 @@ with gr.Blocks(css="footer{display:none !important}", theme=theme) as demo:
|
|
219 |
minimum=1,
|
220 |
maximum=20,
|
221 |
step=0.1,
|
222 |
-
value=5
|
223 |
)
|
224 |
num_inference_steps_base = gr.Slider(
|
225 |
label="Number of inference steps for base",
|
226 |
minimum=10,
|
227 |
maximum=100,
|
228 |
step=1,
|
229 |
-
value=
|
230 |
)
|
231 |
with gr.Row(visible=False) as refiner_params:
|
232 |
guidance_scale_refiner = gr.Slider(
|
|
|
32 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
|
33 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
|
34 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
35 |
+
ENABLE_REFINER = os.getenv("ENABLE_REFINER", "0")#
|
36 |
|
37 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
38 |
if torch.cuda.is_available():
|
|
|
219 |
minimum=1,
|
220 |
maximum=20,
|
221 |
step=0.1,
|
222 |
+
value=7.5,
|
223 |
)
|
224 |
num_inference_steps_base = gr.Slider(
|
225 |
label="Number of inference steps for base",
|
226 |
minimum=10,
|
227 |
maximum=100,
|
228 |
step=1,
|
229 |
+
value=60,
|
230 |
)
|
231 |
with gr.Row(visible=False) as refiner_params:
|
232 |
guidance_scale_refiner = gr.Slider(
|