ising-transformer / ising.py
bertran-yorro's picture
Initial upload: model, training scripts, Gradio app, data
5c85f22 verified
raw
history blame contribute delete
580 Bytes
"""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()