---
license: mit
language:
- en
library_name: diffusers
---


```python

from diffusers import CVVAEModel, AutoencoderKL, UNet3DConditionModelAdapter, ControlNetModel3D, ControlNetModel
import torch
import time

device = torch.device("cuda:0")

vae = CVVAEModel.from_pretrained(
    "/data2/onkar/cvvae/CV-VAE",
    subfolder='vae3d',
    torch_dtype=torch.float16
)
vae.to(device)
print(vae)
vae.requires_grad_(False)

```