Spaces:
Running
on
Zero
Running
on
Zero
remove cache
Browse files
app.py
CHANGED
|
@@ -7,8 +7,9 @@ import numpy as np
|
|
| 7 |
from PIL import Image, ImageOps
|
| 8 |
import os
|
| 9 |
import transformers
|
|
|
|
| 10 |
transformers.utils.move_cache()
|
| 11 |
-
|
| 12 |
pipe = StableDiffusion3InstructPix2PixPipeline.from_pretrained("BleachNick/SD3_UltraEdit_w_mask", torch_dtype=torch.float16).to("cuda")
|
| 13 |
|
| 14 |
|
|
|
|
| 7 |
from PIL import Image, ImageOps
|
| 8 |
import os
|
| 9 |
import transformers
|
| 10 |
+
from transformers.utils.hub import move_cache
|
| 11 |
transformers.utils.move_cache()
|
| 12 |
+
move_cache()
|
| 13 |
pipe = StableDiffusion3InstructPix2PixPipeline.from_pretrained("BleachNick/SD3_UltraEdit_w_mask", torch_dtype=torch.float16).to("cuda")
|
| 14 |
|
| 15 |
|