Update app.py
Browse files
app.py
CHANGED
|
@@ -224,7 +224,7 @@ if not torch.cuda.is_available():
|
|
| 224 |
DESCRIPTION += "\n<p>你现在运行在CPU上 但是此项目只支持GPU.</p>"
|
| 225 |
|
| 226 |
MAX_SEED = np.iinfo(np.int32).max
|
| 227 |
-
MAX_IMAGE_SIZE =
|
| 228 |
|
| 229 |
if torch.cuda.is_available():
|
| 230 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
|
@@ -342,14 +342,14 @@ with gr.Blocks(css=css) as demo:
|
|
| 342 |
minimum=512,
|
| 343 |
maximum=MAX_IMAGE_SIZE,
|
| 344 |
step=64,
|
| 345 |
-
value=
|
| 346 |
)
|
| 347 |
height = gr.Slider(
|
| 348 |
label="高度",
|
| 349 |
minimum=512,
|
| 350 |
maximum=MAX_IMAGE_SIZE,
|
| 351 |
step=64,
|
| 352 |
-
value=
|
| 353 |
)
|
| 354 |
with gr.Row():
|
| 355 |
guidance_scale = gr.Slider(
|
|
|
|
| 224 |
DESCRIPTION += "\n<p>你现在运行在CPU上 但是此项目只支持GPU.</p>"
|
| 225 |
|
| 226 |
MAX_SEED = np.iinfo(np.int32).max
|
| 227 |
+
MAX_IMAGE_SIZE = 1536
|
| 228 |
|
| 229 |
if torch.cuda.is_available():
|
| 230 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
|
|
|
| 342 |
minimum=512,
|
| 343 |
maximum=MAX_IMAGE_SIZE,
|
| 344 |
step=64,
|
| 345 |
+
value=832,
|
| 346 |
)
|
| 347 |
height = gr.Slider(
|
| 348 |
label="高度",
|
| 349 |
minimum=512,
|
| 350 |
maximum=MAX_IMAGE_SIZE,
|
| 351 |
step=64,
|
| 352 |
+
value=1216,
|
| 353 |
)
|
| 354 |
with gr.Row():
|
| 355 |
guidance_scale = gr.Slider(
|