OLM Chat 1B

Environment

cd olm-chat-1b
pip install -r requirements.txt

Usage

python3 scripts/generate.py  --checkpoint logs/open_lm_alpaca_padding_mod1752/checkpoints/epoch_2.pt --positional-embedding-type rotary --input-text '{"instruction":"Using the provided data, what is the most common pet in this household?","input":"The household has 3 cats, 2 dogs, and 1 rabbit."}'