feature: Hugging Face Transformers 4.17 for TF 2.6 (aws#3027)

Co-authored-by: Navin Soni <[email protected]>

Author: Qingzi-Lan

src/sagemaker/image_uri_config/huggingface.json

@@ -561,7 +561,8 @@
             },
             "4.17.0": {
                 "version_aliases": {
-                    "pytorch1.10": "pytorch1.10.2"
+                    "pytorch1.10": "pytorch1.10.2",
+                    "tensorflow2.6": "tensorflow2.6.3"
                 },
                 "pytorch1.10.2": {
                     "py_versions": ["py38"],
@@ -594,6 +595,38 @@
                     },
                     "repository": "huggingface-pytorch-training",
                     "container_version": {"gpu": "cu113-ubuntu20.04"}
+                },
+                "tensorflow2.6.3": {
+                    "py_versions": ["py38"],
+                    "registries": {
+                        "af-south-1": "626614931356",
+                        "ap-east-1": "871362719292",
+                        "ap-northeast-1": "763104351884",
+                        "ap-northeast-2": "763104351884",
+                        "ap-northeast-3": "364406365360",
+                        "ap-south-1": "763104351884",
+                        "ap-southeast-1": "763104351884",
+                        "ap-southeast-2": "763104351884",
+                        "ca-central-1": "763104351884",
+                        "cn-north-1": "727897471807",
+                        "cn-northwest-1": "727897471807",
+                        "eu-central-1": "763104351884",
+                        "eu-north-1": "763104351884",
+                        "eu-south-1": "692866216735",
+                        "eu-west-1": "763104351884",
+                        "eu-west-2": "763104351884",
+                        "eu-west-3": "763104351884",
+                        "me-south-1": "217643126080",
+                        "sa-east-1": "763104351884",
+                        "us-east-1": "763104351884",
+                        "us-east-2": "763104351884",
+                        "us-gov-west-1": "442386744353",
+                        "us-iso-east-1": "886529160074",
+                        "us-west-1": "763104351884",
+                        "us-west-2": "763104351884"
+                    },
+                    "repository": "huggingface-tensorflow-training",
+                    "container_version": {"gpu": "cu112-ubuntu20.04"}
                 }
             }
         }
@@ -989,7 +1022,8 @@
             },
             "4.17.0": {
                 "version_aliases": {
-                    "pytorch1.10": "pytorch1.10.2"
+                    "pytorch1.10": "pytorch1.10.2",
+                    "tensorflow2.6": "tensorflow2.6.3"
                 },
                 "pytorch1.10.2": {
                     "py_versions": ["py38"],
@@ -1022,6 +1056,38 @@
                     },
                     "repository": "huggingface-pytorch-inference",
                     "container_version": {"gpu": "cu113-ubuntu20.04", "cpu": "ubuntu20.04" }
+                },
+                "tensorflow2.6.3": {
+                    "py_versions": ["py38"],
+                    "registries": {
+                        "af-south-1": "626614931356",
+                        "ap-east-1": "871362719292",
+                        "ap-northeast-1": "763104351884",
+                        "ap-northeast-2": "763104351884",
+                        "ap-northeast-3": "364406365360",
+                        "ap-south-1": "763104351884",
+                        "ap-southeast-1": "763104351884",
+                        "ap-southeast-2": "763104351884",
+                        "ca-central-1": "763104351884",
+                        "cn-north-1": "727897471807",
+                        "cn-northwest-1": "727897471807",
+                        "eu-central-1": "763104351884",
+                        "eu-north-1": "763104351884",
+                        "eu-south-1": "692866216735",
+                        "eu-west-1": "763104351884",
+                        "eu-west-2": "763104351884",
+                        "eu-west-3": "763104351884",
+                        "me-south-1": "217643126080",
+                        "sa-east-1": "763104351884",
+                        "us-east-1": "763104351884",
+                        "us-east-2": "763104351884",
+                        "us-gov-west-1": "442386744353",
+                        "us-iso-east-1": "886529160074",
+                        "us-west-1": "763104351884",
+                        "us-west-2": "763104351884"
+                    },
+                    "repository": "huggingface-tensorflow-inference",
+                    "container_version": {"gpu": "cu112-ubuntu20.04", "cpu": "ubuntu20.04" }
                 }
             }
         }

tests/conftest.py

@@ -266,7 +266,7 @@ def huggingface_pytorch_latest_inference_py_version(huggingface_inference_pytorc
 
 @pytest.fixture(scope="module")
 def huggingface_tensorflow_latest_training_py_version():
-    return "py37"
+    return "py38"
 
 
 @pytest.fixture(scope="module")