yuvaranianandhan24 commited on
Commit
b3146b6
·
verified ·
1 Parent(s): 19a195c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def txt2img(text, style="realistic"):
73
  # Use the Euler scheduler here instead
74
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
75
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
76
- pipe = pipe.to("cuda")
77
  image = pipe(prompt = text, guidance_scale = 7.5).images[0]
78
  return image
79
 
 
73
  # Use the Euler scheduler here instead
74
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
75
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
76
+ #pipe = pipe.to("cuda")
77
  image = pipe(prompt = text, guidance_scale = 7.5).images[0]
78
  return image
79