lint + fixes

1986smalltiger · Jun 23, 2024 · 0d234f9 · 0d234f9
1 parent 2c1868e
commit 0d234f9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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(