Devticks commited on
Commit
bb6bed5
·
verified ·
1 Parent(s): 008b67a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ if image is not None:
61
 
62
  model_id = "timbrooks/instruct-pix2pix"
63
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
64
- pipe.to("cuda")
65
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
66
 
67
  images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
 
61
 
62
  model_id = "timbrooks/instruct-pix2pix"
63
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
64
+
65
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
66
 
67
  images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images