mlx-examples/llms/hf_llm/models
Anchen a2402116ae
refactor(hf_llm): moving phi2 example into hf_llm (#293)
* refactor: moving phi2 example into hf_llm

* chore: clean up

* chore: update phi2 model args so it can load args from config

* fix phi2 + nits + readme

* allow any HF repo, update README

* fix bug in llama

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-11 12:29:12 -08:00
..
__init__.py refactor(hf_llm): moving phi2 example into hf_llm (#293) 2024-01-11 12:29:12 -08:00
.gitignore refactor(hf_llm): moving phi2 example into hf_llm (#293) 2024-01-11 12:29:12 -08:00
base.py refactor(hf_llm): moving phi2 example into hf_llm (#293) 2024-01-11 12:29:12 -08:00
llama.py refactor(hf_llm): moving phi2 example into hf_llm (#293) 2024-01-11 12:29:12 -08:00
phi2.py refactor(hf_llm): moving phi2 example into hf_llm (#293) 2024-01-11 12:29:12 -08:00