winglian commited on
Commit
70157cc
1 Parent(s): 3a99495

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: ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
 
 
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)