mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-15 02:28:24 +06:00
ac26260436
* cast image features to model.dtype where needed to support FP16 or other precision in pipelines * Update src/transformers/pipelines/image_feature_extraction.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Use .to instead * Add FP16 pipeline support for zeroshot audio classification * Remove unused torch imports * Add docs on FP16 pipeline * Remove unused import * Add FP16 tests to pipeline mixin * Add fp16 placeholder for mask_generation pipeline test * Add FP16 tests for all pipelines * Fix formatting * Remove torch_dtype arg from is_pipeline_test_to_skip* * Fix format * trigger ci --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| agent.md | ||
| backbones.md | ||
| callback.md | ||
| configuration.md | ||
| data_collator.md | ||
| deepspeed.md | ||
| feature_extractor.md | ||
| image_processor.md | ||
| keras_callbacks.md | ||
| logging.md | ||
| model.md | ||
| onnx.md | ||
| optimizer_schedules.md | ||
| output.md | ||
| pipelines.md | ||
| processors.md | ||
| quantization.md | ||
| text_generation.md | ||
| tokenizer.md | ||
| trainer.md | ||