Views
No views yet
1
2from diffusers import CVVAEModel, AutoencoderKL, UNet3DConditionModelAdapter, ControlNetModel3D, ControlNetModel
3import torch
4import time
5
6device = torch.device("cuda:0")
7
8vae = CVVAEModel.from_pretrained(
9 "/data2/onkar/cvvae/CV-VAE",
10 subfolder='vae3d',
11 torch_dtype=torch.float16
12)
13vae.to(device)
14print(vae)
15vae.requires_grad_(False)
16