gabehubner commited on
Commit
775aae5
·
verified ·
1 Parent(s): 68f7fb8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -10
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
- input_tensor = torch.randn(1, 3, 256, 256) # Replace with your actual input
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`)