Skip to content

Commit 04f66b7

Browse files
committed
[bugfix] fix margin (#5817)
1 parent c8a0049 commit 04f66b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/template/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def _rlhf_encode(self, inputs: TemplateInputs) -> Dict[str, Any]:
369369
data = locals()[f'{prefix}_encoded']
370370
for k, v in data.items():
371371
encoded[f'{prefix}_{k}'] = v
372-
if margin:
372+
if margin is not None:
373373
encoded['margin'] = float(margin)
374374
return encoded
375375

0 commit comments

Comments
 (0)