Add Aimv2 model #36625

yaswanth19
github-actions
github-actions github-actions marked this pull request as draft 1 year ago
yaswanth19
yaswanth19
yaswanth19
yonigozlan
yonigozlan
yaswanth19
yonigozlan
yaswanth19 Model skelton
8198d498
yaswanth19 changes
c63b292a
yaswanth19 temp push
58edea82
yaswanth19 changes
429e5b63
yaswanth19 Added support for aimv2-native
daac3380
yaswanth19 More changes
20f43aab
yaswanth19 More changes
4a0b4424
yaswanth19 yaswanth19 force pushed from a54cbca4 to 4a0b4424 1 year ago
yaswanth19 Stupid mistake correction
11ed21d6
yaswanth19 Added config and refactor
999be2a1
yaswanth19 Added vison model
21b9231b
yaswanth19 update
f59ac3b8
yaswanth19 Refactor for lit variant
c22841e9
yaswanth19 Added Text Model
ce81e5eb
yaswanth19 Minor fixes
2af7afb7
yaswanth19 nits
16f0b928
yaswanth19 update
544c19c9
yaswanth19 Preliminary tests
99f6e5ee
yaswanth19 More fixes
cf4a128c
yaswanth19 Updated tests 🤗
be7490af
yaswanth19 Refactor
b893bc87
yaswanth19 Updated testcase
da7bb612
yaswanth19 Merge branch 'main' into add-aimv2-model
fa814b02
yaswanth19 Updated config
c91ff5ad
yaswanth19 make fixup
104296a3
yaswanth19 more fixes
019210c9
yaswanth19 Bug fix and updates
a031299d
yaswanth19 deadcode
923f76f2
yaswanth19 Fixes
3c2d124a
yaswanth19 nit
4d9b3968
yaswanth19 up
12b20673
yaswanth19 Happy CI ✅
ebd1c9c8
yaswanth19 Reduce LOC
ff5f4d82
yaswanth19 nit
13ce5c83
yaswanth19 nit
54d0d95e
yaswanth19 yaswanth19 force pushed from dcba18d3 to 54d0d95e 360 days ago
yaswanth19 Merge branch 'main' into add-aimv2-model
efdca36c
yaswanth19 make style
9f3fd333
yaswanth19 return_dict refactor
aabc4602
yaswanth19 bug fix
8f85bf8b
yaswanth19 fix
4fdb728d
yaswanth19 doc update
b316999e
yaswanth19 Merge branch 'main' into add-aimv2-model
26d034c8
yaswanth19 yaswanth19 marked this pull request as ready for review 359 days ago
github-actions github-actions requested a review from ArthurZucker ArthurZucker 359 days ago
github-actions github-actions requested a review from Rocketknight1 Rocketknight1 359 days ago
yaswanth19
yaswanth19 yaswanth19 changed the title [WiP] Add Aimv2 model Add Aimv2 model 358 days ago
yaswanth19 Merge branch 'main' into add-aimv2-model
462504e8
yaswanth19 nit
ec510c33
yaswanth19 Merge branch 'main' into add-aimv2-model
a1792294
yaswanth19 Merge branch 'main' into add-aimv2-model
ab22ef60
yaswanth19 make fixup
f2d47eb0
yaswanth19 Minor update
ce67e589
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
f74a9add
yaswanth19 Merge branch 'main' into add-aimv2-model
097cda6d
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
705f45da
yaswanth19 _init_weigths modifcation
67ed7118
yonigozlan
yaswanth19
yaswanth19
yaswanth19 commented on 2025-04-18
yonigozlan
yonigozlan commented on 2025-04-22
yaswanth19 update tests
6277203f
yaswanth19 Minor fixes post review
9af37647
yaswanth19 Merge branch 'main' into add-aimv2-model
76089774
yaswanth19 Update w.r.t GradientCheckpointingLayer
705fa3c7
yaswanth19 yaswanth19 requested a review from yonigozlan yonigozlan 337 days ago
yaswanth19
yaswanth19 docs update
b7e9236c
yonigozlan
yonigozlan commented on 2025-04-23
yaswanth19 update
ef9f9081
yaswanth19 nit
cae9b802
yaswanth19
yaswanth19 commented on 2025-04-23
yaswanth19
yaswanth19 yaswanth19 requested a review from yonigozlan yonigozlan 337 days ago
yonigozlan
yonigozlan commented on 2025-04-28
ArthurZucker ArthurZucker added New model
ArthurZucker
ArthurZucker approved these changes on 2025-04-30
yaswanth19 Merge branch 'main' into add-aimv2-model
b0539477
yaswanth19 Use more Modular 😉
b7802dee
yaswanth19 Change name from AIMv2 to Aimv2
cfd01868
yaswanth19 Nit
9aae4d47
yaswanth19
yaswanth19
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
03679fa4
ArthurZucker
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
367597cc
yaswanth19 Merge branch 'main' into add-aimv2-model
41c5d6aa
yaswanth19 make style
bc5858d5
yaswanth19 Merge branch 'main' into add-aimv2-model
b9c11d75
yaswanth19
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
1a11e86d
ArthurZucker
ArthurZucker
yaswanth19 Add model doc pointer
5bc37096
yaswanth19 Merge branch 'main' into add-aimv2-model
6160836d
yaswanth19 Merge branch 'main' into add-aimv2-model
638af182
yaswanth19 make style
13966059
yaswanth19 Update model doc section
164d0e32
yaswanth19 Merge branch 'main' into add-aimv2-model
c89b2002
yaswanth19
Cyrilvallez
Cyrilvallez commented on 2025-06-17
yaswanth19 Merge branch 'main' into add-aimv2-model
0331fb36
yaswanth19 updates
95506b5a
yaswanth19
yaswanth19
Cyrilvallez
Cyrilvallez
Cyrilvallez approved these changes on 2025-06-25
yaswanth19 Merge branch 'main' into add-aimv2-model
ee091687
yaswanth19 Modify attn mask and interface
dad9d00c
yaswanth19 update test
781834a9
yaswanth19 Merge branch 'main' into add-aimv2-model
89f2c063
yaswanth19
yaswanth19 Merge branch 'main' into add-aimv2-model
ec27afb4
yaswanth19 Final change
8eac21d9
Cyrilvallez Merge branch 'main' into add-aimv2-model
d4261482
Cyrilvallez
Cyrilvallez approved these changes on 2025-06-27
yaswanth19 Utilize flash and flex attn
39ecc8d6
yaswanth19
Cyrilvallez
yaswanth19 keep attn mask
0627ef68
yaswanth19 camelcase model name in test file
446df75b
yaswanth19 Merge branch 'main' into add-aimv2-model
d7d23b9d
yaswanth19 Merge branch 'main' into add-aimv2-model
2af4fac9
github-actions
yaswanth19 Merge branch 'main' into add-aimv2-model
33c2846c
github-actions
yaswanth19 Fix docstring
f0a00cd4
yaswanth19 Fix config warning finally and create_causal_mask
792b34e8
github-actions
yaswanth19 disable torchscript
5fc34600
github-actions
Cyrilvallez remove unused arg
6d3227f7
Cyrilvallez Merge branch 'main' into add-aimv2-model
39380617
github-actions
Cyrilvallez
github-actions
github-actions
yaswanth19
Cyrilvallez remove from tests
9e223034
github-actions
Cyrilvallez balance model size for tests
8d127bd5
github-actions
Cyrilvallez fix device
969148d2
github-actions
Cyrilvallez
github-actions
yaswanth19
Cyrilvallez tests
8467e18a
github-actions
Cyrilvallez tests
4b2ccdc7
Cyrilvallez
github-actions
github-actions
Cyrilvallez flaky test
9899e350
github-actions
Cyrilvallez fix import
0e3720ca
github-actions
Cyrilvallez
Cyrilvallez approved these changes on 2025-07-08
Cyrilvallez Cyrilvallez merged fbdaa7b0 into main 262 days ago
yaswanth19
nahidalam
yaswanth19

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone