Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 101 | logplain |
-rw-r--r-- | attention.py | 4389 | logplain |
d--------- | embeddings | 153 | logplain |
-rw-r--r-- | layers.py | 3481 | logplain |
-rw-r--r-- | local_attention.py | 4653 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 1025 | logplain |
-rw-r--r-- | residual.py | 218 | logplain |
-rw-r--r-- | transformer.py | 1822 | logplain |
-rw-r--r-- | vit.py | 1360 | logplain |