mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
80a421a54d
Temporarily pin 1.26.0 to get the workflow working while I go sort out which dependencies need to be updated Succeeding run: https://github.com/pytorch/pytorch/actions/runs/9877733366/job/27280052419?pr=130442 Tested by adding my branch to the trust relationship for the policy and removing the environment Pull Request resolved: https://github.com/pytorch/pytorch/pull/130442 Approved by: https://github.com/atalman, https://github.com/malfet
22 lines
374 B
Bash
22 lines
374 B
Bash
#!/bin/bash
|
|
|
|
set -euxo pipefail
|
|
|
|
# Download requirements
|
|
cd llm-target-determinator
|
|
pip install -q -r requirements.txt
|
|
cd ../codellama
|
|
pip install -e .
|
|
pip install numpy==1.26.0
|
|
|
|
# Run indexer
|
|
cd ../llm-target-determinator
|
|
|
|
torchrun \
|
|
--standalone \
|
|
--nnodes=1 \
|
|
--nproc-per-node=1 \
|
|
indexer.py \
|
|
--experiment-name indexer-files \
|
|
--granularity FILE
|