Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
diffusers/tests
History
Nicolas Patry 5755d16868
[Proposal] Support loading from safetensors if file is present. (#1357) 
* [Proposal] Support loading from safetensors if file is present.

* Style.

* Fix.

* Adding some test to check loading logic.

+ modify download logic to not download pytorch file if not necessary.

* Fixing the logic.

* Adressing comments.

* factor out into a function.

* Remove dead function.

* Typo.

* Extra fetch only if safetensors is there.

* Apply suggestions from code review

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
 		2022-11-28 10:39:42 +01:00
..
fixtures/custom_pipeline [Custom pipeline] Easier loading of local pipelines (#1327) 2022-11-17 16:00:26 +01:00
models update unet2d (#1376) 2022-11-23 20:46:30 +01:00
pipelines Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-11-25 15:15:09 +00:00
repo_utils Move accelerate to a soft-dependency (#1134) 2022-11-04 14:58:52 +01:00
__init__.py fix issues with loading, add test for pipeline 2022-06-07 15:40:36 +02:00
conftest.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_config.py Deprecate `predict_epsilon` (#1393) 2022-11-25 14:02:15 +01:00
test_layers_utils.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_modeling_common.py Allow to set config params directly in init (#1419) 2022-11-25 15:07:09 +01:00
test_modeling_common_flax.py Allow to set config params directly in init (#1419) 2022-11-25 15:07:09 +01:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py [Proposal] Support loading from safetensors if file is present. (#1357) 2022-11-28 10:39:42 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py Make height and width optional (#1401) 2022-11-24 18:23:59 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py Allow to set config params directly in init (#1419) 2022-11-25 15:07:09 +01:00
test_scheduler_flax.py Allow to set config params directly in init (#1419) 2022-11-25 15:07:09 +01:00
test_training.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_utils.py Deprecate sample size (#1406) 2022-11-24 22:32:44 +01:00