Skip to content

Commit 53e2f23

Browse files
committed
fix
1 parent e4d66d0 commit 53e2f23

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/models/deepseek_vl_hybrid/test_modeling_deepseek_vl_hybrid.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ class DeepseekVLHybridModelTest(ModelTesterMixin, GenerationTesterMixin, unittes
167167
else {}
168168
)
169169
_is_composite = True
170+
model_split_percents = [0.5, 0.85, 0.9] # it tries to offload everything with the default value
170171

171172
def setUp(self):
172173
self.model_tester = DeepseekVLHybridModelTester(self)

0 commit comments

Comments
 (0)