Spaces:
Sleeping
Sleeping
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()
|