Spaces:
Running
on
Zero
Running
on
Zero
Add spaces
Browse files
app.py
CHANGED
@@ -86,7 +86,7 @@ if is_attention_slicing_enabled:
|
|
86 |
if is_cpu_offload_enabled:
|
87 |
pipeline.enable_model_cpu_offload()
|
88 |
|
89 |
-
|
90 |
def remove(gradio_image, rm_guidance_scale=9, num_inference_steps=50, seed=42, strength=0.8):
|
91 |
try:
|
92 |
generator = torch.Generator('cuda').manual_seed(seed)
|
|
|
86 |
if is_cpu_offload_enabled:
|
87 |
pipeline.enable_model_cpu_offload()
|
88 |
|
89 |
+
@spaces.GPU
|
90 |
def remove(gradio_image, rm_guidance_scale=9, num_inference_steps=50, seed=42, strength=0.8):
|
91 |
try:
|
92 |
generator = torch.Generator('cuda').manual_seed(seed)
|