K00B404 commited on
Commit
df82e36
·
verified ·
1 Parent(s): 50a7778

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
6
 
7
  model_id = "timbrooks/instruct-pix2pix"
8
- pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
9
  pipe.to("cpu")
10
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
11
 
 
5
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
6
 
7
  model_id = "timbrooks/instruct-pix2pix"
8
+ pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None)
9
  pipe.to("cpu")
10
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
11