Hugging Face's logo

Spaces:
gpt3
/
analogy
Runtime error

App Files Community
1
ritwikbiswas commited on
Commit
6b993b7
•
1 Parent(s): 84b6163

stable diff no cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  model_id = "stabilityai/stable-diffusion-2-1"
8
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
9
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
10
- pipe = pipe.to("cuda")
11
 
12
  openai.api_key = os.getenv("OPENAI_API_KEY")
13
 
 
7
  model_id = "stabilityai/stable-diffusion-2-1"
8
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
9
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
10
+ # pipe = pipe.to("cuda")
11
 
12
  openai.api_key = os.getenv("OPENAI_API_KEY")
13