mirror of https://github.com/hpcaitech/ColossalAI
parent
c06208e72c
commit
c2c8c9cf17
|
@ -91,7 +91,7 @@ jobs:
|
||||||
container:
|
container:
|
||||||
image: hpcaitech/pytorch-cuda:2.1.0-12.1.0
|
image: hpcaitech/pytorch-cuda:2.1.0-12.1.0
|
||||||
options: --gpus all --rm -v /dev/shm -v /data/scratch/llama-tiny:/data/scratch/llama-tiny
|
options: --gpus all --rm -v /dev/shm -v /data/scratch/llama-tiny:/data/scratch/llama-tiny
|
||||||
timeout-minutes: 75
|
timeout-minutes: 90
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
|
@ -140,7 +140,7 @@ jobs:
|
||||||
|
|
||||||
- name: Install Colossal-AI
|
- name: Install Colossal-AI
|
||||||
run: |
|
run: |
|
||||||
pip install -v -e .
|
BUILD_EXT=1 pip install -v -e .
|
||||||
pip install -r requirements/requirements-test.txt
|
pip install -r requirements/requirements-test.txt
|
||||||
|
|
||||||
- name: Store Colossal-AI Cache
|
- name: Store Colossal-AI Cache
|
||||||
|
|
Loading…
Reference in New Issue