Spaces:

WAT-ai-AA
/

stable-diffused-adversarial-attacks

Runtime error

harshp3030 commited on Feb 9, 2023

Commit

c7923e8

1 Parent(s): 338ad64

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 model_id = "stabilityai/stable-diffusion-2"
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
 def txt2img(prompt):
   image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0]

 model_id = "stabilityai/stable-diffusion-2"
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16")
 def txt2img(prompt):
   image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0]