Update adapter_model.py

adapter_model.py

@@ -22,6 +22,8 @@ DEVICE = str(os.getenv(
     "DEVICE")
     )
 
+
+
 noise_scheduler = DDIMScheduler(
     num_train_timesteps=1000,
     beta_start=0.00085,
@@ -33,6 +35,10 @@ noise_scheduler = DDIMScheduler(
 )
 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)
+
 
 class MODEL:
     def __init__(self, action):