Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3a104e9
1
Parent(s):
eb3568a
modify to support zeroGPU
Browse files
app.py
CHANGED
|
@@ -91,7 +91,6 @@ def prune_model(pipe, hookers):
|
|
| 91 |
ffn_hook.clear_hooks()
|
| 92 |
return pipe
|
| 93 |
|
| 94 |
-
@spaces.GPU
|
| 95 |
def binary_mask_eval(args):
|
| 96 |
# load sdxl model
|
| 97 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
|
|
| 91 |
ffn_hook.clear_hooks()
|
| 92 |
return pipe
|
| 93 |
|
|
|
|
| 94 |
def binary_mask_eval(args):
|
| 95 |
# load sdxl model
|
| 96 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|