trying VIT model

[sweep-ai[bot]]

Apply Sweep Rules to your PR?

• [ ] Apply: Leftover TODOs in the code should be handled.
• [ ] Apply: All new business logic should have corresponding unit tests in the tests/ directory.
• [ ] Apply: Any clearly inefficient or repeated code should be optimized or refactored.