Hugging Face's logo

Spaces:
JoPmt
/
Txt2Img_Seg_SSD_1B
Runtime error

App Files Community
JoPmt commited on
Commit
578daf0
·
1 Parent(s): bb643a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from diffusers import StableDiffusionXLPipeline
7
  accelerator = Accelerator(cpu=True)
8
 
9
  pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.bfloat16, use_safetensors=True, variant="fp16"))
 
10
  pipe.to("cpu")
11
 
12
  def plex(prompt,neg_prompt,stips):
 
7
  accelerator = Accelerator(cpu=True)
8
 
9
  pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.bfloat16, use_safetensors=True, variant="fp16"))
10
+ pipe.unet.to(memory_format=torch.channels_last)
11
  pipe.to("cpu")
12
 
13
  def plex(prompt,neg_prompt,stips):