Skip to content

Commit 4b5659d

Browse files
authored
Revert "change: add TGI 2.4.0 image uri (#4922)" (#4926)
1 parent dc98ace commit 4b5659d

File tree

3 files changed

+1
-49
lines changed

3 files changed

+1
-49
lines changed

CONTRIBUTING.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ Before sending us a pull request, please ensure that:
7676
1. Install tox using `pip install tox`
7777
1. Install coverage using `pip install .[test]`
7878
1. cd into the sagemaker-python-sdk folder: `cd sagemaker-python-sdk` or `cd /environment/sagemaker-python-sdk`
79-
1. Run the following tox command and verify that all code checks and unit tests pass: `tox -- tests/unit`
79+
1. Run the following tox command and verify that all code checks and unit tests pass: `tox tests/unit`
8080
1. You can also run a single test with the following command: `tox -e py310 -- -s -vv <path_to_file><file_name>::<test_function_name>`
8181
1. You can run coverage via runcvoerage env : `tox -e runcoverage -- tests/unit` or `tox -e py310 -- tests/unit --cov=sagemaker --cov-append --cov-report xml`
8282
* Note that the coverage test will fail if you only run a single test, so make sure to surround the command with `export IGNORE_COVERAGE=-` and `unset IGNORE_COVERAGE`

src/sagemaker/image_uri_config/huggingface-llm.json

Lines changed: 0 additions & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -766,53 +766,6 @@
766766
"container_version": {
767767
"gpu": "cu124-ubuntu22.04"
768768
}
769-
},
770-
"2.4.0": {
771-
"py_versions": [
772-
"py311"
773-
],
774-
"registries": {
775-
"af-south-1": "626614931356",
776-
"il-central-1": "780543022126",
777-
"ap-east-1": "871362719292",
778-
"ap-northeast-1": "763104351884",
779-
"ap-northeast-2": "763104351884",
780-
"ap-northeast-3": "364406365360",
781-
"ap-south-1": "763104351884",
782-
"ap-south-2": "772153158452",
783-
"ap-southeast-1": "763104351884",
784-
"ap-southeast-2": "763104351884",
785-
"ap-southeast-3": "907027046896",
786-
"ap-southeast-4": "457447274322",
787-
"ca-central-1": "763104351884",
788-
"cn-north-1": "727897471807",
789-
"cn-northwest-1": "727897471807",
790-
"eu-central-1": "763104351884",
791-
"eu-central-2": "380420809688",
792-
"eu-north-1": "763104351884",
793-
"eu-west-1": "763104351884",
794-
"eu-west-2": "763104351884",
795-
"eu-west-3": "763104351884",
796-
"eu-south-1": "692866216735",
797-
"eu-south-2": "503227376785",
798-
"me-south-1": "217643126080",
799-
"me-central-1": "914824155844",
800-
"sa-east-1": "763104351884",
801-
"us-east-1": "763104351884",
802-
"us-east-2": "763104351884",
803-
"us-gov-east-1": "446045086412",
804-
"us-gov-west-1": "442386744353",
805-
"us-iso-east-1": "886529160074",
806-
"us-isob-east-1": "094389454867",
807-
"us-west-1": "763104351884",
808-
"us-west-2": "763104351884",
809-
"ca-west-1": "204538143572"
810-
},
811-
"tag_prefix": "2.4.0-tgi2.4.0",
812-
"repository": "huggingface-pytorch-tgi-inference",
813-
"container_version": {
814-
"gpu": "cu124-ubuntu22.04"
815-
}
816769
}
817770
}
818771
}

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@
4646
"2.0.2": "2.3.0-tgi2.0.2-gpu-py310-cu121-ubuntu22.04",
4747
"2.2.0": "2.3.0-tgi2.2.0-gpu-py310-cu121-ubuntu22.04-v2.0",
4848
"2.3.1": "2.4.0-tgi2.3.1-gpu-py311-cu124-ubuntu22.04",
49-
"2.4.0": "2.4.0-tgi2.4.0-gpu-py311-cu124-ubuntu22.04",
5049
},
5150
"inf2": {
5251
"0.0.16": "1.13.1-optimum0.0.16-neuronx-py310-ubuntu22.04",

0 commit comments

Comments
 (0)