Skip to content

Commit e799516

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

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

src/diffusers/utils/testing_utils.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1201,10 +1201,3 @@ def update_mapping_from_spec(device_fn_dict: Dict[str, Callable], attribute_name
12011201
update_mapping_from_spec(BACKEND_MAX_MEMORY_ALLOCATED, "MAX_MEMORY_ALLOCATED_FN")
12021202

12031203

1204-
@require_torch
1205-
def flush_memory(device: str, gc_collect=False, reset_mem_stats=False):
1206-
if gc_collect:
1207-
gc.collect()
1208-
if reset_mem_stats:
1209-
backend_reset_peak_memory(device)
1210-
backend_empty_cache(device)

0 commit comments

Comments
 (0)