Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove duplicate log_time(...) function.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-21 18:10:24 +00:00 committed by Brandon
parent 72398350b4
commit ac96f187bd
1 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/backend/quantization/load_flux_model_bnb_llm_int8.py
View File

@ -1,5 +1,3 @@
import time
from contextlib import contextmanager
from pathlib import Path from pathlib import Path
import accelerate import accelerate
@ -7,17 +5,7 @@ from diffusers.models.transformers.transformer_flux import FluxTransformer2DMode
from safetensors.torch import load_file, save_file from safetensors.torch import load_file, save_file
from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8 from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8
from invokeai.backend.quantization.load_flux_model_bnb_nf4 import log_time
@contextmanager
def log_time(name: str):
"""Helper context manager to log the time taken by a block of code."""
start = time.time()
try:
yield None
finally:
end = time.time()
print(f"'{name}' took {end - start:.4f} secs")
def main(): def main():