OLM Chat 7B

Environment

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

Usage

python3 scripts/generate.py  --model open_lm_7b --checkpoint logs/test_alpaca_7b_1p25_240612/checkpoints/epoch_1.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."}'