Update fcdm_diffae/model.py benchmark and docs
Browse files- fcdm_diffae/model.py +1 -1
fcdm_diffae/model.py
CHANGED
|
@@ -47,7 +47,7 @@ class FCDMDiffAE(nn.Module):
|
|
| 47 |
"""Standalone FCDM DiffAE model for HuggingFace distribution.
|
| 48 |
|
| 49 |
A diffusion autoencoder built on FCDM (Fully Convolutional Diffusion Model)
|
| 50 |
-
blocks. Encodes images to compact
|
| 51 |
VP-parameterized diagonal Gaussian posterior, and decodes them back via
|
| 52 |
iterative VP diffusion with a skip-concat decoder.
|
| 53 |
|
|
|
|
| 47 |
"""Standalone FCDM DiffAE model for HuggingFace distribution.
|
| 48 |
|
| 49 |
A diffusion autoencoder built on FCDM (Fully Convolutional Diffusion Model)
|
| 50 |
+
blocks. Encodes images to compact 384-channel spatial latents via a
|
| 51 |
VP-parameterized diagonal Gaussian posterior, and decodes them back via
|
| 52 |
iterative VP diffusion with a skip-concat decoder.
|
| 53 |
|