ritwikbiswas commited on
Commit
42cd0c7
1 Parent(s): b29e29f

try diffusion fix

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,8 +5,9 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
5
  import torch
6
 
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")
 
5
  import torch
6
 
7
  model_id = "stabilityai/stable-diffusion-2-1"
8
+ pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
9
+ # pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
10
+ # pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
11
  # pipe = pipe.to("cuda")
12
 
13
  openai.api_key = os.getenv("OPENAI_API_KEY")