Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ vae = VAE(
|
|
| 32 |
out_ch=3,
|
| 33 |
ch_mult=[1, 2, 4, 4],
|
| 34 |
num_res_blocks=2,
|
| 35 |
-
|
| 36 |
).cuda().bfloat16()
|
| 37 |
|
| 38 |
from safetensors.torch import load_file
|
|
|
|
| 32 |
out_ch=3,
|
| 33 |
ch_mult=[1, 2, 4, 4],
|
| 34 |
num_res_blocks=2,
|
| 35 |
+
z_channels=16
|
| 36 |
).cuda().bfloat16()
|
| 37 |
|
| 38 |
from safetensors.torch import load_file
|