File size: 423 Bytes
bb3719f 5c4b0eb 6819709 5c4b0eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
---
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)
``` |