Zhenhong commited on
Commit
bc392de
1 Parent(s): fc09f75

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  model_id = "runwayml/stable-diffusion-v1-5"
6
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
7
- pipe = pipe.to("cuda")
8
 
9
  # prompt = "a photo of an astronaut riding a horse on mars"
10
  # image.save("astronaut_rides_horse.png")
 
4
 
5
  model_id = "runwayml/stable-diffusion-v1-5"
6
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
7
+ #pipe = pipe.to("cuda")
8
 
9
  # prompt = "a photo of an astronaut riding a horse on mars"
10
  # image.save("astronaut_rides_horse.png")