PR #42684 adds jais2 model support

adds jais2 model support

sarathc-cerebras committed 315 days ago

updates tests

sarathc-cerebras committed 315 days ago

addresses review comment

sarathc-cerebras committed 315 days ago

review comments addressed

sarathc-cerebras committed 315 days ago

addresses test review comments

sarathc-cerebras committed 315 days ago

fixes date

sarathc-cerebras committed 315 days ago

format issue fix

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/__init__.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

Update tests/models/jais2/test_modeling_jais2.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras committed 315 days ago

fixes tests as per review comment

sarathc-cerebras committed 315 days ago

updates layernorm setup

sarathc-cerebras committed 315 days ago

Apply suggestions from code review

sarathc-cerebras committed 315 days ago

addressed review comments and updated tests as recomended

sarathc-cerebras committed 315 days ago

fixup tests

vasqu committed 315 days ago

Merge branch 'main' into add-jais2-model

vasqu committed 314 days ago

transformers adds jais2 model support #42684 Merged