PR #35610 Refactor (and fix) gpt_neox

Refactor (and fix) gpt_neox #35610

Cyrilvallez merged 29 commits into main from still-more-refactors

vasqu commented on 2025-01-10

start a nice modular

04b4e9e1

Update modular_gpt_neox.py

7f1def94

Update modular_gpt_neox.py

fe5d1430

Update modular_gpt_neox.py

29d6c824

Update modular_gpt_neox.py

d1550956

update

331338d9

Update modular_gpt_neox.py

55879833

convert

f77f219f

fix attribute

cac76068

fix attrs

1df569b4

oups

4f68d6ab

fix

fbc88309

fix

67a36109

fix

74d100f0

fix

fb1856cc

fix

ff7488c8

fix order to pass test (see with accelerate team)

6e69d92d

trigger CIs

1b300952

modular

6a735a1a

Cyrilvallez force pushed from 900f5534 to 6a735a1a 1 year ago

update

79763b8c

aec220c7

Update test_modeling_gpt_neox.py

5f65ae33

Update test_modeling_gpt_neox.py

b8ea1682

trigger CIs

9085156c

correctly pass arg

05fd6b25

Cyrilvallez changed the title ~~Still more model refactors!~~ Refactor (and fix) gpt_neox 1 year ago

Cyrilvallez marked this pull request as ready for review 1 year ago

ArthurZucker commented on 2025-02-03

simplify

7bb2070e

remove key warning

a0ebe8de

update tp -> it's compatible since the view is before

a72d9072

ArthurZucker commented on 2025-02-04

ArthurZucker approved these changes on 2025-02-04

trigger CIs

d585e72b

Cyrilvallez merged 9afb904b into main 1 year ago

Cyrilvallez deleted the still-more-refactors branch 1 year ago

Reviewers

ArthurZucker

vasqu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Refactor (and fix) gpt_neox #35610 Merged

Refactor (and fix) gpt_neox #35610

transformers
Refactor (and fix) gpt_neox
#35610

Merged