huggingface/accelerate

Pull Requests Commits

SunMarc committed 2 years ago

c8f6f791

fix case num_process=1

SunMarc committed 2 years ago

7ca4bccf

Store split points in hf_split_points

muellerzr committed 2 years ago

8792a8c5

Put split_points in pipelien

muellerzr committed 2 years ago

e3f6b99b

Use no split module classes explicitly

muellerzr committed 2 years ago

df7779aa

muellerzr committed 2 years ago

77f8e92b

muellerzr committed 2 years ago

449eb8d9

muellerzr committed 2 years ago

9eef9dd4

Working version!

muellerzr committed 2 years ago

06f04a99

Timing I would expect

muellerzr committed 2 years ago

2767bb19

muellerzr committed 2 years ago

e713e28e

Just 40* (#2332)

muellerzr committed 2 years ago

Verified 31fd2b1a

Fixed typos in readme files of docs folder. (#2329)

rishit5 committed 2 years ago

Verified fce61a99

Fix model memory issue (#2327)

muellerzr committed 2 years ago

Verified 6ec92cf0

convert it back to dict (#2326)

SunMarc committed 2 years ago

Verified 2a403732

Raise error when using batches of different sizes with `dispatch_batches=True` (#2325)

SunMarc committed 2 years ago

Verified f823404f

Update versions to dev

muellerzr committed 2 years ago

ef2fe912

Fix infer_auto_device_map when tied weights share the same prefix name (#2324)

fxmarty committed 2 years ago

Verified e3e9b875

`Params4bit` added to bnb classes in set_module_tensor_to_device() (#2315)

poedator committed 2 years ago

Verified 456afd92

fix sanity check (#2310)

izhx committed 2 years ago

Verified 0d2280da

Bring old seed technique back (#2319)

muellerzr committed 2 years ago

Verified 55d4a496

Update test_deepspeed.py (#2323)

pacman100 committed 2 years ago

Verified 2a8829d9

Fix DeepSpeed related regression (#2304)

pacman100 committed 2 years ago

Verified 3969731c

DeepSpeed refactoring (#2313)

pacman100 committed 2 years ago

Verified 411aa58a

Update accelerator.py (#2295)

stas00 committed 2 years ago

Verified 4420ec64

Update docs: Add warning for device_map=None for load_checkpoint_and_dispatch (#2308)

PhilJd committed 2 years ago

Verified 2241725a

Add more missing items (#2309)

muellerzr committed 2 years ago

Verified 5cac8789

[deepspeed] documentation (#2296)

stas00 committed 2 years ago

Verified 5d314233

make test_state_checkpointing device agnostic (#2290)

ji-huazhong committed 2 years ago

Verified 2721387b

Fix breakpoint API in test_script.py on TPU. (#2263)

vanbasten23 committed 2 years ago

Verified 2cfa88bd

Older