mirror of https://github.com/InternLM/InternLM
update
parent
4f565ac16d
commit
515f87ec02
|
@ -41,7 +41,7 @@ jobs:
|
|||
source activate internlm-model-latest
|
||||
pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118
|
||||
pip install /mnt/petrelfs/qa-caif-cicd/resource/flash_attn-2.5.8+cu118torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
||||
pip install sentencepiece auto-gptq==0.6.0 lmdeploy[all]
|
||||
pip install sentencepiece auto-gptq==0.6.0 lmdeploy[all] beautifulsoup4
|
||||
srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} --gpus-per-task=2 pytest -s -v -m tmp --color=yes ./tests/test_hf_model.py
|
||||
conda deactivate
|
||||
- name: remove_env
|
||||
|
|
Loading…
Reference in New Issue