Skip to content

Commit e814635

Browse files
faaanyhlky
andauthored
Update src/diffusers/utils/testing_utils.py
Co-authored-by: hlky <[email protected]>
1 parent d15618b commit e814635

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/utils/testing_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1128,7 +1128,7 @@ def backend_device_count(device: str):
11281128
return _device_agnostic_dispatch(device, BACKEND_DEVICE_COUNT)
11291129

11301130

1131-
def backend_reset_peak_memory(device: str):
1131+
def backend_reset_peak_memory_stats(device: str):
11321132
return _device_agnostic_dispatch(device, BACKEND_RESET_PEAK_MEMORY_STATS)
11331133

11341134

0 commit comments

Comments
 (0)