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