Skip to content

Commit 55814b1

Browse files
committed
fixed deepspeed bugs
Signed-off-by: ftgreat <ftgreat@163.com>
1 parent 01dfdbd commit 55814b1

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

examples/Aquila/Aquila-chat/Aquila-chat.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ save_rng: True
1313
lora: False
1414
eps: 1.0e-8
1515
enable_sft_dataset_dir: './data/'
16-
enable_sft_dataset_file: 'sft_v0.9.4_train.jsonl'
16+
enable_sft_dataset_file: 'convo_samples.jsonl'

flagai/fp16/fp16.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,8 @@ def forward(self, *inputs, **kwargs):
7979
def named_parameters(self, prefix: str = '', recurse: bool = True):
8080
return self.module.named_parameters(prefix=prefix, recurse=recurse)
8181

82-
def parameters(self):
83-
return self.module.parameters()
82+
def parameters(self, recurse: bool = True):
83+
return self.module.parameters(recurse=recurse)
8484

8585
def state_dict(self, destination=None, prefix='', keep_vars=False):
8686
return self.module.state_dict(destination, prefix, keep_vars)

0 commit comments

Comments
 (0)