Hugging Face's logo

Spaces:
jeasinema
/
UltraEdit-SD3
Running on Zero

App Files Community
jeasinema commited on
Commit
e4b358e
•
1 Parent(s): 60c69d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -6,6 +6,9 @@ import PIL.Image
6
  import numpy as np
7
  from PIL import Image, ImageOps
8
  import os
 
 
 
9
  pipe = StableDiffusion3InstructPix2PixPipeline.from_pretrained("BleachNick/SD3_UltraEdit_w_mask", torch_dtype=torch.float16).to("cuda")
10
 
11
 
 
6
  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