mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-24 14:58:56 +06:00
f53fe35b29
* Draft fast image processors * Draft working fast version * py3.8 compatible cache * Enable loading fast image processors through auto * Tidy up; rescale behaviour based on input type * Enable tests for fast image processors * Smarter rescaling * Don't default to Fast * Safer imports * Add necessary Pillow requirement * Woops * Add AutoImageProcessor test * Fix up * Fix test for imagegpt * Fix test * Review comments * Add warning for TF and JAX input types * Rearrange * Return transforms * NumpyToTensor transformation * Rebase - include changes from upstream in ImageProcessingMixin * Safe typing * Fix up * convert mean/std to tesnor to rescale * Don't store transforms in state * Fix up * Update src/transformers/image_processing_utils_fast.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Update src/transformers/models/auto/image_processing_auto.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Update src/transformers/models/auto/image_processing_auto.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Update src/transformers/models/auto/image_processing_auto.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Warn if fast image processor available * Update src/transformers/models/vit/image_processing_vit_fast.py * Transpose incoming numpy images to be in CHW format * Update mapping names based on packages, auto set fast to None * Fix up * Fix * Add AutoImageProcessor.from_pretrained(checkpoint, use_fast=True) test * Update src/transformers/models/vit/image_processing_vit_fast.py Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com> * Add equivalence and speed tests * Fix up --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
33 lines
447 B
Plaintext
33 lines
447 B
Plaintext
tensorboard
|
|
scikit-learn
|
|
seqeval
|
|
psutil
|
|
sacrebleu >= 1.4.12
|
|
git+https://github.com/huggingface/accelerate@main#egg=accelerate
|
|
rouge-score
|
|
tensorflow_datasets
|
|
matplotlib
|
|
git-python==1.0.3
|
|
faiss-cpu
|
|
streamlit
|
|
elasticsearch
|
|
nltk
|
|
pandas
|
|
datasets >= 1.13.3
|
|
fire
|
|
pytest<8.0.1
|
|
conllu
|
|
sentencepiece != 0.1.92
|
|
protobuf
|
|
torch
|
|
torchvision
|
|
torchaudio
|
|
jiwer
|
|
librosa
|
|
evaluate >= 0.2.0
|
|
timm
|
|
albumentations >= 1.4.5
|
|
torchmetrics
|
|
pycocotools
|
|
Pillow>=10.0.1,<=15.0
|