File size: 580 Bytes
5c85f22
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
"""Legacy entry point.  This file has been split into:
  model.py  — Generator architecture and gen_config
  train.py  — training loop  (python train.py --help)
  sample.py — checkpoint loading and autoregressive sampling  (python sample.py --help)
"""
# Re-export model symbols for any code that still does `from ising import ...`
from model import (  # noqa: F401
    snake_order,
    EmbedderBlock,
    FeedForwardBlock,
    AttentionBlock,
    TransformerLayer,
    Encoder,
    Generator,
    gen_config,
)

if __name__ == "__main__":
    import train
    train.main()