Skip to content

Commit f345f5d

Browse files
duanjunwenflybird11111
authored andcommitted
[fix] fix test_pipeline_utils ci;
1 parent 86d4644 commit f345f5d

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

tests/test_pipeline/test_pipeline_utils/test_t5_pipeline_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ def __init__(self):
1515
self.is_interleave = False
1616
self.num_layers_per_stage = None
1717
self.num_model_chunks = 1
18+
self.use_zbv = False
1819

1920
@property
2021
def num_stages(self):

tests/test_pipeline/test_pipeline_utils/test_whisper_pipeline_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ def __init__(self):
1515
self.is_interleave = False
1616
self.num_layers_per_stage = None
1717
self.num_model_chunks = 1
18+
self.use_zbv = False
1819

1920
@property
2021
def num_stages(self):

0 commit comments

Comments
 (0)