Update README.md
Browse files
README.md
CHANGED
@@ -46,14 +46,5 @@ $$
|
|
46 |
| GAN Start Step | 50,000 |
|
47 |
| KL Annealing | Yes (10% of training) |
|
48 |
| Augmentations | Crop, flip, jitter, blur, rotation |
|
49 |
-
Trained using a cosine learning rate schedule with gradient clipping and automatic mixed precision (`torch.cuda.amp`)
|
50 |
-
## Usage Example
|
51 |
-
```python
|
52 |
-
import torch
|
53 |
-
from diffusers import AutoencoderKL
|
54 |
-
vae = AutoencoderKL.from_pretrained("gabehubner/vae-256px-8z")
|
55 |
-
vae.eval()
|
56 |
|
57 |
-
|
58 |
-
with torch.no_grad():
|
59 |
-
output = vae(input_tensor).sample
|
|
|
46 |
| GAN Start Step | 50,000 |
|
47 |
| KL Annealing | Yes (10% of training) |
|
48 |
| Augmentations | Crop, flip, jitter, blur, rotation |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
|
50 |
+
Trained using a cosine learning rate schedule with gradient clipping and automatic mixed precision (`torch.cuda.amp`)
|
|
|
|