Update adapter_model.py

adapter_model.py

@@ -1,6 +1,6 @@
 from dotenv import load_dotenv
 import os 
-from diffusers import StableDiffusionInpaintPipelineLegacy,StableDiffusionControlNetPipeline,ControlNetModel, DDIMScheduler,AutoencoderKL
+from diffusers import StableDiffusionInpaintPipelineLegacy,StableDiffusionControlNetPipeline,ControlNetModel, DDIMScheduler,ConsistencyDecoderVAE
 import torch
 from ip_adapter import IPAdapter
 
@@ -9,9 +9,9 @@ load_dotenv()
 BASE_MODEL_PATH = str(os.getenv(
     "BASE_MODEL_PATH")
     )
-VAE_MODEL_PATH = str(os.getenv(
-    "VAE_MODEL_PATH")
-    )
+#VAE_MODEL_PATH = str(os.getenv(
+#    "VAE_MODEL_PATH")
+#    )
 IMAGE_ENCODER_PATH = str(os.getenv(
     "IMAGE_ENCODER_PATH")
     )
@@ -33,11 +33,8 @@ noise_scheduler = DDIMScheduler(
     set_alpha_to_one=False,
     steps_offset=1,
 )
-vae = AutoencoderKL.from_pretrained(VAE_MODEL_PATH).to(dtype=torch.float16)
-
-#TODO 
-#from diffusers import ConsistencyDecoderVAE
-#vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
+#vae = AutoencoderKL.from_pretrained(VAE_MODEL_PATH).to(dtype=torch.float16)
+vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
 
 
 class MODEL: