Skip to content

Commit 76af785

Browse files
Copilotanxiangsir
andcommitted
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

File tree

pyproject.toml

Lines changed: 0 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -47,56 +47,6 @@ exclude = [
4747
"pages*",
4848
]
4949

50-
[build-system]
51-
requires = ["setuptools>=61.0"]
52-
build-backend = "setuptools.build_meta"
53-
54-
[project]
55-
name = "onevision-encoder"
56-
version = "0.1.0"
57-
description = "HEVC-Style Vision Transformer for efficient video understanding"
58-
readme = "README.md"
59-
license = {text = "Apache-2.0"}
60-
requires-python = ">=3.8"
61-
dependencies = [
62-
"torch>=2.0.0",
63-
"transformers>=4.40.0",
64-
"timm>=0.9.0",
65-
"numpy",
66-
"Pillow",
67-
]
68-
69-
[project.optional-dependencies]
70-
train = [
71-
"tensorboard",
72-
"nvidia-dali-cuda110",
73-
"decord",
74-
]
75-
dev = [
76-
"pytest",
77-
"black",
78-
"isort",
79-
]
80-
81-
[tool.setuptools.packages.find]
82-
where = ["."]
83-
include = [
84-
"onevision_encoder*",
85-
"dataloader*",
86-
"model_factory*",
87-
"training*",
88-
"dataset*",
89-
"eval_encoder*",
90-
]
91-
exclude = [
92-
"shells*",
93-
"tools*",
94-
"llava_next*",
95-
"asset*",
96-
"pages*",
97-
]
98-
99-
10050
[tool.ruff]
10151
target-version = "py310"
10252
line-length = 119

0 commit comments

Comments
 (0)