Commit
·
e23bf60
1
Parent(s):
9f8624d
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ pipeline.to(device)
|
|
| 16 |
if USE_TORCH_COMPILE:
|
| 17 |
pipeline.unet = torch.compile(pipeline.unet, mode="reduce-overhead", fullgraph=True)
|
| 18 |
|
| 19 |
-
def generate(num_images_per_prompt: int = 1)
|
| 20 |
print("Start...")
|
| 21 |
print("Version", torch.__version__)
|
| 22 |
|
|
|
|
| 16 |
if USE_TORCH_COMPILE:
|
| 17 |
pipeline.unet = torch.compile(pipeline.unet, mode="reduce-overhead", fullgraph=True)
|
| 18 |
|
| 19 |
+
def generate(num_images_per_prompt: int = 1):
|
| 20 |
print("Start...")
|
| 21 |
print("Version", torch.__version__)
|
| 22 |
|