Spaces:

JoPmt
/

Txt2Img_Seg_SSD_1B

Runtime error

JoPmt commited on Dec 20, 2023

Commit

1d407e4

1 Parent(s): 4d7c596

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from accelerate import Accelerator
 from diffusers import StableDiffusionXLPipeline
 accelerator = Accelerator(cpu=True)
-pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.float32, use_safetensors=True, variant="fp32"))
 pipe = accelerator.prepare(pipe.to("cpu"))
 def plex(prompt,neg_prompt,stips):

 from diffusers import StableDiffusionXLPipeline
 accelerator = Accelerator(cpu=True)
+pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.bfloat16, use_safetensors=True, variant="fp16"))
 pipe = accelerator.prepare(pipe.to("cpu"))
 def plex(prompt,neg_prompt,stips):