xref: /aosp_15_r20/external/executorch/examples/models/llama/install_requirements.sh (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1#!/bin/bash
2# Copyright (c) Meta Platforms, Inc. and affiliates.
3# All rights reserved.
4#
5# This source code is licensed under the BSD-style license found in the
6# LICENSE file in the root directory of this source tree.
7
8# Install snakeviz for cProfile flamegraph
9# Install sentencepiece for llama tokenizer
10pip install snakeviz sentencepiece
11
12# Install torchao.
13pip install "$(dirname "$0")/../../../third-party/ao"
14
15# Install lm-eval for Model Evaluation with lm-evalution-harness
16# Install tiktoken for tokenizer
17pip install lm_eval==0.4.5
18pip install tiktoken blobfile
19# Restore numpy if >= 2.0
20pip install "numpy<2.0"
21
22# Call the install helper for further setup
23python examples/models/llama/install_requirement_helper.py
24