Marc Sun 6ea08a90b0 Fix sharding when no device_map is passed (#8531) ...

* Fix sharding when no device_map is passed

* style

* add tests

* align

* add docstring

* format

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2024-06-20 11:05:27 -10:00

..

fixtures

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

lora

Add Stable Diffusion 3 (#8483)

2024-06-12 20:44:00 +01:00

models

Fix sharding when no device_map is passed (#8531)

2024-06-20 11:05:27 -10:00

others

#7535 Update FloatTensor type hints to Tensor (#7883)

2024-05-10 09:53:31 -10:00

pipelines

Add Stable Diffusion 3 (#8483)

2024-06-12 20:44:00 +01:00

schedulers

[scheduler] support custom timesteps and sigmas (#7817)

2024-05-09 11:07:43 -10:00

single_file

[tests] decorate StableDiffusion21PipelineSingleFileSlowTests with slow. (#7941)

2024-05-14 14:26:21 -10:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00