Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,6 +57,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
|
|
| 57 |
pipe.to(device)
|
| 58 |
MAX_SEED = 1000000
|
| 59 |
MAX_PIXEL_BUDGET = 2048 * 2048
|
|
|
|
| 60 |
|
| 61 |
|
| 62 |
def process_input(input_image, upscale_factor, **kwargs):
|
|
|
|
| 57 |
pipe.to(device)
|
| 58 |
MAX_SEED = 1000000
|
| 59 |
MAX_PIXEL_BUDGET = 2048 * 2048
|
| 60 |
+
torch.cuda.empty_cache()
|
| 61 |
|
| 62 |
|
| 63 |
def process_input(input_image, upscale_factor, **kwargs):
|