always print time

This commit is contained in:
lllyasviel
2024-02-23 13:02:30 -08:00
parent df12dde12e
commit ad0ce480f9

View File

@@ -427,9 +427,7 @@ def load_models_gpu(models, memory_required=0):
free_memory(extra_mem, d, models_already_loaded)
moving_time = time.perf_counter() - execution_start_time
if moving_time > 0.1:
print(f'Moving model(s) skipped. Freeing memory has taken {moving_time:.2f} seconds')
print(f'Moving model(s) skipped. Cleanup has taken {moving_time:.2f} seconds')
return
print(f"Begin to load {len(models_to_load)} model{'s' if len(models_to_load) > 1 else ''}")
@@ -476,8 +474,7 @@ def load_models_gpu(models, memory_required=0):
current_loaded_models.insert(0, loaded_model)
moving_time = time.perf_counter() - execution_start_time
if moving_time > 0.1:
print(f'Moving model(s) has taken {moving_time:.2f} seconds')
print(f'Moving model(s) has taken {moving_time:.2f} seconds')
return