diff --git a/libs/infinity_emb/infinity_emb/sync_engine.py b/libs/infinity_emb/infinity_emb/sync_engine.py index 59c430cb..bd80bcc6 100644 --- a/libs/infinity_emb/infinity_emb/sync_engine.py +++ b/libs/infinity_emb/infinity_emb/sync_engine.py @@ -107,7 +107,6 @@ def __init__(self, _engine_args_array: list[EngineArgs]): except Exception as e: self.async_close_loop() raise e - @classmethod def from_args(cls, engine_args_array: list[EngineArgs]) -> "SyncEngineArray": diff --git a/libs/infinity_emb/infinity_emb/transformer/utils_optimum.py b/libs/infinity_emb/infinity_emb/transformer/utils_optimum.py index 616fdaae..91236629 100644 --- a/libs/infinity_emb/infinity_emb/transformer/utils_optimum.py +++ b/libs/infinity_emb/infinity_emb/transformer/utils_optimum.py @@ -76,7 +76,9 @@ def optimize_model( if Path(model_name_or_path).exists() else Path(HUGGINGFACE_HUB_CACHE) / "infinity_onnx" / model_name_or_path ) - files_optimized = list(path_folder.glob(f"**/*{execution_provider}_optimized.onnx.onnx")) + files_optimized = list( + path_folder.glob(f"**/*{execution_provider}_optimized.onnx.onnx") + ) if execution_provider == "TensorrtExecutionProvider": return model_class.from_pretrained( model_name_or_path, @@ -142,7 +144,9 @@ def optimize_model( revision=revision, trust_remote_code=trust_remote_code, provider=execution_provider, - file_name=Path(file_name).name.replace(".onnx", f"{execution_provider}_optimized.onnx"), + file_name=Path(file_name).name.replace( + ".onnx", f"{execution_provider}_optimized.onnx" + ), ) except Exception as e: logger.warning(