Stas Bekman 2f5507580b [from_pretrained] extend torch_dtype="auto" to look up config.torch_dtype first, expand docs (#21524) ...

* [from_pretrained] expand on torch_dtype entry

* fold 4 into 1

* style

* support torch_dtype='config' plus tests

* style

* oops

* fold config into auto, fix bug

* fix check

* better log

* better log

* clean up

2023-02-10 09:09:21 -08:00

..

benchmark

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

deepspeed

[tests] add missing report_to none (#21505)

2023-02-08 09:32:40 -08:00

extended

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

fixtures

[WIP] add SpeechT5 model (#18922)

2023-02-03 12:43:46 -05:00

generation

Generate: TF .generate() can now be exported with dynamic length (#21474)

2023-02-09 12:52:30 +00:00

mixed_int8

[bnb] Fine-tuning HF 8-bit models (#21290)

2023-02-02 16:39:23 +01:00

models

Add X-MOD (#20939)

2023-02-10 15:32:06 +01:00

onnx

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

optimization

Add inverse sqrt learning rate scheduler (#21495)

2023-02-07 15:00:50 -05:00

pipelines

[pipeline] A simple fix for half-precision & 8bit models (#21479)

2023-02-10 10:26:17 +01:00

repo_utils

Cleanup quality (#21493)

2023-02-07 12:27:31 -05:00

sagemaker

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

tokenization

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

trainer

Fix epoch number when resuming training (#21478)

2023-02-06 19:34:34 -05:00

utils

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

__init__.py

GPU text generation: mMoved the encoded_prompt to correct device

2020-01-06 15:11:12 +01:00

test_configuration_common.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_feature_extraction_common.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_image_processing_common.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_image_transforms.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_modeling_common.py

[from_pretrained] extend torch_dtype="auto" to look up config.torch_dtype first, expand docs (#21524)

2023-02-10 09:09:21 -08:00

test_modeling_flax_common.py

[Tests] Improve flax test_attention_outputs (#21486)

2023-02-10 11:31:49 -05:00

test_modeling_tf_common.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_sequence_feature_extraction_common.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_tokenization_common.py

Cleanup quality (#21493)

2023-02-07 12:27:31 -05:00