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`)
|
|
|
|
|
|