add a latest tag for regular axolotl image, cleanup extraneous print statement (#746)
Browse files
.github/workflows/main.yml
CHANGED
|
@@ -23,6 +23,7 @@ jobs:
|
|
| 23 |
python_version: "3.10"
|
| 24 |
pytorch: 2.0.1
|
| 25 |
axolotl_extras:
|
|
|
|
| 26 |
- cuda: 118
|
| 27 |
cuda_version: 11.8.0
|
| 28 |
python_version: "3.10"
|
|
@@ -54,7 +55,9 @@ jobs:
|
|
| 54 |
PYTORCH_VERSION=${{ matrix.pytorch }}
|
| 55 |
file: ./docker/Dockerfile
|
| 56 |
push: ${{ github.event_name != 'pull_request' }}
|
| 57 |
-
tags:
|
|
|
|
|
|
|
| 58 |
labels: ${{ steps.metadata.outputs.labels }}
|
| 59 |
build-axolotl-runpod:
|
| 60 |
needs: build-axolotl
|
|
|
|
| 23 |
python_version: "3.10"
|
| 24 |
pytorch: 2.0.1
|
| 25 |
axolotl_extras:
|
| 26 |
+
is_latest: true
|
| 27 |
- cuda: 118
|
| 28 |
cuda_version: 11.8.0
|
| 29 |
python_version: "3.10"
|
|
|
|
| 55 |
PYTORCH_VERSION=${{ matrix.pytorch }}
|
| 56 |
file: ./docker/Dockerfile
|
| 57 |
push: ${{ github.event_name != 'pull_request' }}
|
| 58 |
+
tags: |
|
| 59 |
+
${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
|
| 60 |
+
${{ (matrix.is_latest) && format('{0}-latest', steps.metadata.outputs.tags) || '' }}
|
| 61 |
labels: ${{ steps.metadata.outputs.labels }}
|
| 62 |
build-axolotl-runpod:
|
| 63 |
needs: build-axolotl
|
src/axolotl/utils/tokenization.py
CHANGED
|
@@ -34,6 +34,5 @@ def check_example_labels(example, tokenizer, text_only=False):
|
|
| 34 |
delimiter = "" if text_only else " "
|
| 35 |
LOG.info(delimiter.join(colored_tokens))
|
| 36 |
LOG.info("\n\n\n")
|
| 37 |
-
print(" ".join(colored_tokens))
|
| 38 |
|
| 39 |
return " ".join(colored_tokens)
|
|
|
|
| 34 |
delimiter = "" if text_only else " "
|
| 35 |
LOG.info(delimiter.join(colored_tokens))
|
| 36 |
LOG.info("\n\n\n")
|
|
|
|
| 37 |
|
| 38 |
return " ".join(colored_tokens)
|