Spaces:
Configuration error
Configuration error
Update adapter_model.py
Browse files- 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
|
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 :
|