Spaces:
Runtime error
Runtime error
fix to 4 resolutions + enable torch.compile
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ if not torch.cuda.is_available():
|
|
| 19 |
MAX_SEED = np.iinfo(np.int32).max
|
| 20 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|
| 21 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
| 22 |
-
USE_TORCH_COMPILE =
|
| 23 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
| 24 |
PREVIEW_IMAGES = True
|
| 25 |
|
|
@@ -152,16 +152,16 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 152 |
with gr.Row():
|
| 153 |
width = gr.Slider(
|
| 154 |
label="Width",
|
| 155 |
-
minimum=
|
| 156 |
maximum=MAX_IMAGE_SIZE,
|
| 157 |
-
step=
|
| 158 |
value=1024,
|
| 159 |
)
|
| 160 |
height = gr.Slider(
|
| 161 |
label="Height",
|
| 162 |
-
minimum=
|
| 163 |
maximum=MAX_IMAGE_SIZE,
|
| 164 |
-
step=
|
| 165 |
value=1024,
|
| 166 |
)
|
| 167 |
num_images_per_prompt = gr.Slider(
|
|
|
|
| 19 |
MAX_SEED = np.iinfo(np.int32).max
|
| 20 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|
| 21 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
| 22 |
+
USE_TORCH_COMPILE = True
|
| 23 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
| 24 |
PREVIEW_IMAGES = True
|
| 25 |
|
|
|
|
| 152 |
with gr.Row():
|
| 153 |
width = gr.Slider(
|
| 154 |
label="Width",
|
| 155 |
+
minimum=1024,
|
| 156 |
maximum=MAX_IMAGE_SIZE,
|
| 157 |
+
step=512,
|
| 158 |
value=1024,
|
| 159 |
)
|
| 160 |
height = gr.Slider(
|
| 161 |
label="Height",
|
| 162 |
+
minimum=1024,
|
| 163 |
maximum=MAX_IMAGE_SIZE,
|
| 164 |
+
step=512,
|
| 165 |
value=1024,
|
| 166 |
)
|
| 167 |
num_images_per_prompt = gr.Slider(
|