Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 27 | logplain |
-rw-r--r-- | attention.py | 4284 | logplain |
d--------- | axial_attention | 158 | logplain |
-rw-r--r-- | decoder.py | 2990 | logplain |
d--------- | embeddings | 150 | logplain |
-rw-r--r-- | layers.py | 3164 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 1141 | logplain |
-rw-r--r-- | residual.py | 245 | logplain |