hassanblend commited on
Commit
01a50ac
1 Parent(s): ec451ae

update i2i pipeline

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -30,7 +30,10 @@ pipe = diffusers.StableDiffusionPipeline.from_pretrained(
30
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
31
  model_id,
32
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
33
- scheduler=scheduler)
 
 
 
34
 
35
  if torch.cuda.is_available():
36
  pipe = pipe.to("cuda")
 
30
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
31
  model_id,
32
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
33
+ scheduler=scheduler,
34
+ vae=vae
35
+ )
36
+
37
 
38
  if torch.cuda.is_available():
39
  pipe = pipe.to("cuda")