Skip to content

[Proposal] Additional Architecture Adapter tests #1302

@jlarson4

Description

@jlarson4

Proposal

Add config tests for remaining architecture adapters

Motivation

Only about a third of our Architecture Adapters have focused tests in tests/unit/model_bridge/supported_architectures/. Use the existing available test suites as examples for how the new tests should be written.

Pitch

Create a small test suite for each of the remaining architecture adapters that do not have one at present. Feel free to tackle as few or as many at once as you'd like, just leave a comment stating which architecture(s) you're working on to reduce duplication of effort.

Architectures that need tests

  • llama
  • mistral
  • mixtral
  • gemma1
  • gemma2
  • gpt2 (@sunny1401 assigned)
  • qwen
  • qwen2 (@Rishik00 assigned)
  • qwen3
  • t5
  • gpt_oss
  • bloom
  • opt (@willytop8 assigned)
  • phi
  • phi3
  • pythia
  • neo
  • neox
  • gptj
  • bert
  • apertus
  • hubert
  • olmo
  • olmo2
  • olmo3
  • olmoe
  • openelm
  • granite
  • granite_moe
  • granite_moe_hybrid
  • qwen3_next
  • stablelm
  • llava_next
  • llava_onevision

Checklist

  • I have checked that there is no similar issue in the repo (required)

Metadata

Metadata

Labels

TransformerBridgeBug specific to the new TransformerBridge systemcomplexity-simpleSimple issues, which may be good for beginnersenhancementNew feature or requestgood first issueGood for newcomershelp wantedExtra attention is neededlow-priorityMaintainers are not prioritising this work currently.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions