Hugging Face's logo

Spaces:
Timtical
/
Character-creation-ai
Sleeping

App Files Community
Timtical commited on
Commit
32bdcc8
·
verified ·
1 Parent(s): bfa8910

Update app.py

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