load_model_with_dolomite_demo.py

import transformers
from dolomite_engine import hf_models

if __name__ == "__main__":
    hf_models = transformers.AutoModelForCausalLM.from_pretrained(
            'shawntan/stickbreaking-3b',
    )