convert-hf : remove einops requirement for InternLM2

This commit is contained in:
Francis Couture-Harpin 2024-05-04 16:52:06 -04:00
parent 0c3833286e
commit 98db4347e8
4 changed files with 22 additions and 20 deletions

View file

@ -86,7 +86,6 @@ let
# TODO(Green-Sky): find a better way to opt-into the heavy ml python runtime
llama-python-extra = python3.withPackages (
ps: [
ps.einops
ps.numpy
ps.sentencepiece
ps.tiktoken