calebaryee321 commited on
Commit
89859a3
1 Parent(s): 6ae7ae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def img_Generation(text):
32
  # Use the Euler scheduler here instead
33
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
34
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
35
- #pipe = pipe.to("cuda")
36
  image = pipe(text, num_inference_steps = 150).images[0]
37
  #image.save("img_1.png")
38
 
 
32
  # Use the Euler scheduler here instead
33
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
34
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
35
+ pipe = pipe.to("cuda")
36
  image = pipe(text, num_inference_steps = 150).images[0]
37
  #image.save("img_1.png")
38