Hugging Face's logo

Spaces:
sbicy
/
text2img
Runtime error

App Files Community
sbicy commited on
Commit
3aaaf76
·
verified ·
1 Parent(s): 8913a43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ import torch
4
 
5
  # Load the Stable Diffusion model
6
  pipe = StableDiffusionPipeline.from_pretrained(
7
- "runwayml/stable-diffusion-v1-5",
8
- torch_dtype=torch.float16
9
  )
 
10
  pipe.to("cuda" if torch.cuda.is_available() else "cpu")
11
 
12
  # Define the image generation function
 
4
 
5
  # Load the Stable Diffusion model
6
  pipe = StableDiffusionPipeline.from_pretrained(
7
+ "runwayml/stable-diffusion-v1-5"
 
8
  )
9
+
10
  pipe.to("cuda" if torch.cuda.is_available() else "cpu")
11
 
12
  # Define the image generation function