Spaces:
Running on Zero
Running on Zero
update app
Browse files
app.py
CHANGED
|
@@ -165,7 +165,7 @@ def update_dimensions_on_upload(image):
|
|
| 165 |
|
| 166 |
return new_width, new_height
|
| 167 |
|
| 168 |
-
@spaces.GPU
|
| 169 |
def infer(
|
| 170 |
images,
|
| 171 |
prompt,
|
|
@@ -254,7 +254,7 @@ def infer(
|
|
| 254 |
gc.collect()
|
| 255 |
torch.cuda.empty_cache()
|
| 256 |
|
| 257 |
-
@spaces.GPU
|
| 258 |
def infer_example(images, prompt, lora_adapter):
|
| 259 |
if not images:
|
| 260 |
return None, 0
|
|
|
|
| 165 |
|
| 166 |
return new_width, new_height
|
| 167 |
|
| 168 |
+
@spaces.GPU(size="xlarge")
|
| 169 |
def infer(
|
| 170 |
images,
|
| 171 |
prompt,
|
|
|
|
| 254 |
gc.collect()
|
| 255 |
torch.cuda.empty_cache()
|
| 256 |
|
| 257 |
+
@spaces.GPU(size="xlarge")
|
| 258 |
def infer_example(images, prompt, lora_adapter):
|
| 259 |
if not images:
|
| 260 |
return None, 0
|