How to use sumitrwk/my-stateOnlyLM with Transformers:
# Load model directly from transformers import StateLM model = StateLM.from_pretrained("sumitrwk/my-stateOnlyLM", dtype="auto")
