altayavci commited on
Commit
193be6a
1 Parent(s): 6f14603

Update adapter_model.py

Browse files
Files changed (1) hide show
  1. adapter_model.py +2 -1
adapter_model.py CHANGED
@@ -1,6 +1,6 @@
1
  from dotenv import load_dotenv
2
  import os
3
- from diffusers import StableDiffusionInpaintPipelineLegacy, StableDiffusionControlNetPipeline, ControlNetModel, DDIMScheduler, AutoencoderKL, ConsistencyDecoderVAE
4
  import torch
5
  from ip_adapter import IPAdapter
6
 
@@ -37,6 +37,7 @@ noise_scheduler = DDIMScheduler(
37
  )
38
 
39
  if OPENAI_CONSISTENCY_VAE == "ENABLE":
 
40
  vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
41
 
42
  else :
 
1
  from dotenv import load_dotenv
2
  import os
3
+ from diffusers import StableDiffusionInpaintPipelineLegacy, StableDiffusionControlNetPipeline, ControlNetModel, DDIMScheduler, AutoencoderKL
4
  import torch
5
  from ip_adapter import IPAdapter
6
 
 
37
  )
38
 
39
  if OPENAI_CONSISTENCY_VAE == "ENABLE":
40
+ from diffusers import ConsistencyDecoderVAE
41
  vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
42
 
43
  else :