Skip to content

Commit f8c75ad

Browse files
authored
[bugfix] fix profiling patch (#4915)
1 parent 1e8727f commit f8c75ad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/trainers/rlhf_trainer/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def patch_profiling_decorator(func):
155155
@functools.wraps(func)
156156
def wrapper(self, *args, **kwargs):
157157
with patch_profiling_context(self, func.__name__):
158-
return func(*args, **kwargs)
158+
return func(self, *args, **kwargs)
159159

160160
return wrapper
161161

0 commit comments

Comments
 (0)