Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
ff1a4e2
1
Parent(s):
1c7c461
remove xformers
Browse files- app_masked.py +1 -1
- requirements.txt +1 -2
app_masked.py
CHANGED
|
@@ -30,7 +30,7 @@ basepipeline = DiffusionPipeline.from_pretrained(
|
|
| 30 |
|
| 31 |
basepipeline = basepipeline.to(DEVICE)
|
| 32 |
|
| 33 |
-
basepipeline.
|
| 34 |
|
| 35 |
@spaces.GPU(duration=30)
|
| 36 |
def image_to_image(
|
|
|
|
| 30 |
|
| 31 |
basepipeline = basepipeline.to(DEVICE)
|
| 32 |
|
| 33 |
+
basepipeline.enable_model_cpu_offload()
|
| 34 |
|
| 35 |
@spaces.GPU(duration=30)
|
| 36 |
def image_to_image(
|
requirements.txt
CHANGED
|
@@ -9,5 +9,4 @@ spaces
|
|
| 9 |
sentencepiece
|
| 10 |
controlnet_aux
|
| 11 |
peft
|
| 12 |
-
tqdm
|
| 13 |
-
xformers
|
|
|
|
| 9 |
sentencepiece
|
| 10 |
controlnet_aux
|
| 11 |
peft
|
| 12 |
+
tqdm
|
|
|