| .. |
|
benchmark
|
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization (#15725)
|
1 year ago |
|
bettertransformer
|
3042c63a95
Add methods to PreTrainedModel to use PyTorch's BetterTransformer (#21259)
|
1 month ago |
|
bitsandbytes
|
9d73b92269
4-bit QLoRA via bitsandbytes (4-bit base model + LoRA) (#23479)
|
3 days ago |
|
deepspeed
|
fe1f5a639d
Fix decorator order (#22708)
|
1 month ago |
|
extended
|
1306b7d3ae
[tests] switch to torchrun (#22712)
|
1 month ago |
|
fixtures
|
e4bacf6614
[WIP] add SpeechT5 model (#18922)
|
3 months ago |
|
generation
|
f8b2574416
Better TF docstring types (#23477)
|
3 days ago |
|
models
|
d8222be57e
[Whisper] Reduce batch size in tests (#23736)
|
3 days ago |
|
onnx
|
81a73fa638
Fix issue introduced in PR #23163 (#23363)
|
1 week ago |
|
optimization
|
8e5a1b2abb
Make schedulers picklable by making lr_lambda fns global (#21768)
|
2 months ago |
|
pipelines
|
2f424d7979
[image-to-text pipeline] Add conditional text support + GIT (#23362)
|
5 days ago |
|
repo_utils
|
6e4bc67099
Revamp test selection for the example tests (#23737)
|
2 days ago |
|
sagemaker
|
5427250351
Avoid invalid escape sequences, use raw strings (#22936)
|
1 month ago |
|
tokenization
|
6f79d26442
Update quality tooling for formatting (#21480)
|
3 months ago |
|
tools
|
4eea25b445
Fix image segmentation tool test (#23306)
|
2 weeks ago |
|
trainer
|
796162c512
Paged Optimizer + Lion Optimizer for Trainer (#23217)
|
3 days ago |
|
utils
|
89159651ba
Fix the regex in `get_imports` to support multiline try blocks and excepts with specific exception types (#23725)
|
3 days ago |
|
__init__.py
|
81d6841b4b
GPU text generation: mMoved the encoded_prompt to correct device
|
3 years ago |
|
test_backbone_common.py
|
90e8263d91
Add methods to update and verify out_features out_indices (#23031)
|
3 weeks ago |
|
test_configuration_common.py
|
b4d4d6fe87
Add RWKV-4 (#22797)
|
2 weeks ago |
|
test_feature_extraction_common.py
|
74c55ab9e5
Prepare tests for hfh 0.14 (#22958)
|
1 month ago |
|
test_image_processing_common.py
|
74c55ab9e5
Prepare tests for hfh 0.14 (#22958)
|
1 month ago |
|
test_image_transforms.py
|
154c6bb7ac
Rescale image back if it was scaled during PIL conversion (#22458)
|
1 month ago |
|
test_modeling_common.py
|
bbbc5c15d4
[AutoModel] fix `torch_dtype=auto` in `from_pretrained` (#23379)
|
1 week ago |
|
test_modeling_flax_common.py
|
88399476c3
Fix bigbird random attention (#21023)
|
1 month ago |
|
test_modeling_tf_common.py
|
814de8fac7
Overhaul TF serving signatures + dummy inputs (#23234)
|
3 days ago |
|
test_pipeline_mixin.py
|
ffad4f1373
Update tiny models and pipeline tests (#23446)
|
1 week ago |
|
test_sequence_feature_extraction_common.py
|
5e8c8eb5ba
Apply ruff flake8-comprehensions (#21694)
|
3 months ago |
|
test_tokenization_common.py
|
0c65fb7cfa
chore: allow protobuf 3.20.3 requirement (#22759)
|
2 weeks ago |
Sylvain Gugger
6e4bc67099
Revamp test selection for the example tests (#23737)