How to use max-rl/maze_17x17_sft_bs256_checkpoints with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("max-rl/maze_17x17_sft_bs256_checkpoints", dtype="auto")
