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