From 234e94e9ddf357b3d94fa0bac7030691601cb482 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 11 Nov 2024 15:09:29 -0500 Subject: [PATCH] replace references to personal docker hub to org docker hub (#2036) [skip ci] --- README.md | 8 ++++---- cicd/Dockerfile.jinja | 2 +- docker/Dockerfile | 2 +- docker/Dockerfile-cloud | 2 +- docker/Dockerfile-cloud-no-tmux | 2 +- docker/Dockerfile-tests | 2 +- docs/debugging.qmd | 6 +++--- 7 files changed, 12 insertions(+), 12 deletions(-) diff --git a/README.md b/README.md index 077dd6fee..7e4d2e376 100644 --- a/README.md +++ b/README.md @@ -159,7 +159,7 @@ accelerate launch -m axolotl.cli.train https://raw.githubusercontent.com/axolotl #### Docker ```bash - docker run --gpus '"all"' --rm -it winglian/axolotl:main-latest + docker run --gpus '"all"' --rm -it axolotlai/axolotl:main-latest ``` Or run on the current files for development: @@ -178,7 +178,7 @@ accelerate launch -m axolotl.cli.train https://raw.githubusercontent.com/axolotl A more powerful Docker command to run would be this: ```bash -docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name axolotl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="${PWD}",target=/workspace/axolotl -v ${HOME}/.cache/huggingface:/root/.cache/huggingface winglian/axolotl:main-latest +docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name axolotl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="${PWD}",target=/workspace/axolotl -v ${HOME}/.cache/huggingface:/root/.cache/huggingface axolotlai/axolotl:main-latest ``` It additionally: @@ -210,7 +210,7 @@ docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name axolotl -- #### Cloud GPU -For cloud GPU providers that support docker images, use [`winglian/axolotl-cloud:main-latest`](https://hub.docker.com/r/winglian/axolotl-cloud/tags) +For cloud GPU providers that support docker images, use [`axolotlai/axolotl-cloud:main-latest`](https://hub.docker.com/r/axolotlai/axolotl-cloud/tags) - on Latitude.sh use this [direct link](https://latitude.sh/blueprint/989e0e79-3bf6-41ea-a46b-1f246e309d5c) - on JarvisLabs.ai use this [direct link](https://jarvislabs.ai/templates/axolotl) @@ -319,7 +319,7 @@ Write a job description in YAML as below: # dstack.yaml type: task -image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.2 +image: axolotlai/axolotl-cloud:main-latest env: - HUGGING_FACE_HUB_TOKEN diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja index 8ce655005..8fe84ac64 100644 --- a/cicd/Dockerfile.jinja +++ b/cicd/Dockerfile.jinja @@ -1,4 +1,4 @@ -FROM winglian/axolotl-base:{{ BASE_TAG }} +FROM axolotlai/axolotl-base:{{ BASE_TAG }} ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX" ENV AXOLOTL_EXTRAS="{{ AXOLOTL_EXTRAS }}" diff --git a/docker/Dockerfile b/docker/Dockerfile index 4872b3907..6e14a70a7 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,5 +1,5 @@ ARG BASE_TAG=main-base -FROM winglian/axolotl-base:$BASE_TAG +FROM axolotlai/axolotl-base:$BASE_TAG ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX" ARG AXOLOTL_EXTRAS="" diff --git a/docker/Dockerfile-cloud b/docker/Dockerfile-cloud index c0bb266d2..d7e3277d2 100644 --- a/docker/Dockerfile-cloud +++ b/docker/Dockerfile-cloud @@ -1,5 +1,5 @@ ARG BASE_TAG=main -FROM winglian/axolotl:$BASE_TAG +FROM axolotlai/axolotl:$BASE_TAG ENV HF_DATASETS_CACHE="/workspace/data/huggingface-cache/datasets" ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub" diff --git a/docker/Dockerfile-cloud-no-tmux b/docker/Dockerfile-cloud-no-tmux index 3e59d4119..6dfea4677 100644 --- a/docker/Dockerfile-cloud-no-tmux +++ b/docker/Dockerfile-cloud-no-tmux @@ -1,5 +1,5 @@ ARG BASE_TAG=main -FROM winglian/axolotl:$BASE_TAG +FROM axolotlai/axolotl:$BASE_TAG ENV HF_DATASETS_CACHE="/workspace/data/huggingface-cache/datasets" ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub" diff --git a/docker/Dockerfile-tests b/docker/Dockerfile-tests index 4a946372d..36c0a639e 100644 --- a/docker/Dockerfile-tests +++ b/docker/Dockerfile-tests @@ -1,5 +1,5 @@ ARG BASE_TAG=main-base -FROM winglian/axolotl-base:$BASE_TAG +FROM axolotlai/axolotl-base:$BASE_TAG ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX" ARG AXOLOTL_EXTRAS="" diff --git a/docs/debugging.qmd b/docs/debugging.qmd index 029549d85..d119dce18 100644 --- a/docs/debugging.qmd +++ b/docs/debugging.qmd @@ -185,7 +185,7 @@ style="border-radius: 10px; display: block; margin: auto;" width="560" height="3 ## Debugging With Docker -Using [official Axolotl Docker images](https://hub.docker.com/r/winglian/axolotl/tags) is a great way to debug your code, and is a very popular way to use Axolotl. Attaching VSCode to Docker takes a few more steps. +Using [official Axolotl Docker images](https://hub.docker.com/r/axolotlai/axolotl/tags) is a great way to debug your code, and is a very popular way to use Axolotl. Attaching VSCode to Docker takes a few more steps. ### Setup @@ -202,11 +202,11 @@ cd axolotl Next, run the desired docker image and mount the current directory. Below is a docker command you can run to do this:[^2] ```bash -docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name axolotl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="${PWD}",target=/workspace/axolotl -v ${HOME}/.cache/huggingface:/root/.cache/huggingface winglian/axolotl:main-py3.10-cu118-2.0.1 +docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name axolotl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="${PWD}",target=/workspace/axolotl -v ${HOME}/.cache/huggingface:/root/.cache/huggingface axolotlai/axolotl:main-py3.10-cu118-2.0.1 ``` >[!Tip] -> To understand which containers are available, see the [Docker section of the README](../README.md#docker) and the [DockerHub repo](https://hub.docker.com/r/winglian/axolotl/tags). For details of how the Docker containers are built, see axolotl's [Docker CI builds](../.github/workflows/main.yml). +> To understand which containers are available, see the [Docker section of the README](../README.md#docker) and the [DockerHub repo](https://hub.docker.com/r/axolotlai/axolotl/tags). For details of how the Docker containers are built, see axolotl's [Docker CI builds](../.github/workflows/main.yml). You will now be in the container. Next, perform an editable install of Axolotl: