Downloads · 30 days
0
onkarsus13/CVVAE_Modified
CVVAE_Modified is a machine learning model from onkarsus13. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as mit.
Downloads · 30 days
0
Access
Public
Updated May 28, 2025
Repo size
1.6 GB
Likes
0
Public
Click a slice to open those files.
.ckpt905 MB · 55%
From the Hugging Face model README
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)