onkarsus13 commited on
Commit
bb3719f
·
verified ·
1 Parent(s): 9dadd91

Update README.md

Browse files

```python3

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)

```

Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -1,3 +1,6 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ library_name: diffusers
6
+ ---