github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-26 12:14:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] fix test_pipeline_utils ci;

Browse Source
This commit is contained in:
duanjunwen
2024-09-26 07:18:16 +00:00
parent 45f17fc6cc
commit c5503b0d80
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_pipeline/test_pipeline_utils/test_t5_pipeline_utils.py
View File

@@ -15,6 +15,7 @@ class _PipelineStageManager(PipelineStageManager):
self.is_interleave = False self.is_interleave = False
self.num_layers_per_stage = None self.num_layers_per_stage = None
self.num_model_chunks = 1 self.num_model_chunks = 1
self.use_zbv = False
@property @property
def num_stages(self): def num_stages(self):

1
tests/test_pipeline/test_pipeline_utils/test_whisper_pipeline_utils.py
View File

@@ -15,6 +15,7 @@ class _PipelineStageManager(PipelineStageManager):
self.is_interleave = False self.is_interleave = False
self.num_layers_per_stage = None self.num_layers_per_stage = None
self.num_model_chunks = 1 self.num_model_chunks = 1
self.use_zbv = False
@property @property
def num_stages(self): def num_stages(self):