Spaces:
Running
on
Zero
Running
on
Zero
Clean up
Browse files
app.py
CHANGED
@@ -54,9 +54,6 @@ if torch.cuda.is_available():
|
|
54 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
55 |
if ENABLE_REFINER:
|
56 |
refiner.unet = torch.compile(refiner.unet, mode="reduce-overhead", fullgraph=True)
|
57 |
-
else:
|
58 |
-
pipe = None
|
59 |
-
refiner = None
|
60 |
|
61 |
|
62 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
|
54 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
55 |
if ENABLE_REFINER:
|
56 |
refiner.unet = torch.compile(refiner.unet, mode="reduce-overhead", fullgraph=True)
|
|
|
|
|
|
|
57 |
|
58 |
|
59 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|