Upload fcdm_diffae/encoder.py with huggingface_hub
Browse files- fcdm_diffae/encoder.py +1 -1
fcdm_diffae/encoder.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
"""
|
| 2 |
|
| 3 |
No input RMSNorm (use_other_outer_rms_norms=False during training).
|
| 4 |
Post-bottleneck RMSNorm (affine=False) on the mean branch.
|
|
|
|
| 1 |
+
"""FCDM DiffAE encoder: patchify -> FCDMBlocks -> diagonal Gaussian posterior.
|
| 2 |
|
| 3 |
No input RMSNorm (use_other_outer_rms_norms=False during training).
|
| 4 |
Post-bottleneck RMSNorm (affine=False) on the mean branch.
|