Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 22 | logplain |
-rw-r--r-- | conv_transformer.py | 4489 | logplain |
d--------- | encoders | 78 | logplain |
d--------- | quantizer | 191 | logplain |
d--------- | transformer | 261 | logplain |
-rw-r--r-- | util.py | 977 | logplain |
-rw-r--r-- | vq_transformer.py | 2674 | logplain |
d--------- | vqvae | 262 | logplain |