src/sagemaker_huggingface_inference_toolkit/diffusers_utils.py (1 line): - line 54: # TODO: add support for more images (Reason is correct output) setup.py (1 line): - line 63: # TODO: Remove upper bound of TF 2.11 once transformers release contains this fix: https://github.com/huggingface/evaluate/pull/372