Travis Johnson d8663cb8c5 Fix bugs in mllama image processing (#36156) ... * fix: handle input_channel_dim == channels_last Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> * fix: default PIL images to channels_last Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> * Apply suggestions from code review Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com> * fixup from review batch Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> * test: add 1x1 PIL image to ambiguous channel test Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> * fix(mllama): avoid 0 dimension for image with impractical aspect ratio Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> --------- Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com> Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>		2025-03-11 10:22:48 +01:00
..
agents
bettertransformer
deepspeed
extended
fixtures
fsdp
generation
models	Fix bugs in mllama image processing (#36156)	2025-03-11 10:22:48 +01:00
optimization
peft_integration
pipelines
quantization
repo_utils
sagemaker
tensor_parallel
tokenization
trainer
utils
__init__.py
test_backbone_common.py
test_configuration_common.py
test_feature_extraction_common.py
test_image_processing_common.py
test_image_transforms.py
test_modeling_common.py
test_modeling_flax_common.py
test_modeling_tf_common.py
test_pipeline_mixin.py
test_processing_common.py
test_sequence_feature_extraction_common.py
test_tokenization_common.py
test_training_args.py