Feature: Add new Hugging Face DLCs to Image config

[philschmid]

Add Release Image to Image config for the Hugging Face DLCs

huggingface_pytorch

Training:

• PT 1.9 GPU, transformers==4.12

inference:

• PT 1.9 CPU, transformers==4.12
• PT 1.9 GPU, transformers==4.12

huggingface_tensorflow

Training:

• TF 2.4 GPU, transformers==4.12
• TF 2.5 GPU, transformers==4.12

inference:

• TF 2.5 CPU, transformers==4.12
• TF 2.5 GPU, transformers==4.12

cc @ahsan-z-khan @arjkesh

[philschmid]

PR open #2752.

The Release issue in the Deep Learning Container repository can be found here:

• [huggingface_tensorflow, huggingface_pytorch] Release Images for Transformers 4.12 deep-learning-containers#1475

[ahsan-z-khan]

@philschmid is there any due date on this?

[philschmid]

Yes, we need to have a SDK version support those images by Wednesday. I already opened the PR and made all required changes to the mage_uri_config

• feature: Hugging Face Transformers 4.12 for Pt1.9/TF2.5  #2752