Commit 76af785
Add test for AutoModel vs OneVisionEncoderModel output consistency
- Created tests/test_automodel_consistency.py with tests for:
- Output consistency between AutoModel and OneVisionEncoderModel
- Flash attention 2 and eager attention implementations
- Batch input handling
- bfloat16 dtype consistency
- Parametrized tests for transformers versions 5.0.0rc1 and 4.53.1
- Fixed duplicate section in pyproject.toml that was preventing pytest from running
Co-authored-by: anxiangsir <31175974+anxiangsir@users.noreply.github.com>1 parent 668b0bd commit 76af785
2 files changed
Lines changed: 413 additions & 50 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
47 | 47 | | |
48 | 48 | | |
49 | 49 | | |
50 | | - | |
51 | | - | |
52 | | - | |
53 | | - | |
54 | | - | |
55 | | - | |
56 | | - | |
57 | | - | |
58 | | - | |
59 | | - | |
60 | | - | |
61 | | - | |
62 | | - | |
63 | | - | |
64 | | - | |
65 | | - | |
66 | | - | |
67 | | - | |
68 | | - | |
69 | | - | |
70 | | - | |
71 | | - | |
72 | | - | |
73 | | - | |
74 | | - | |
75 | | - | |
76 | | - | |
77 | | - | |
78 | | - | |
79 | | - | |
80 | | - | |
81 | | - | |
82 | | - | |
83 | | - | |
84 | | - | |
85 | | - | |
86 | | - | |
87 | | - | |
88 | | - | |
89 | | - | |
90 | | - | |
91 | | - | |
92 | | - | |
93 | | - | |
94 | | - | |
95 | | - | |
96 | | - | |
97 | | - | |
98 | | - | |
99 | | - | |
100 | 50 | | |
101 | 51 | | |
102 | 52 | | |
| |||
0 commit comments