diff --git a/docs/references/api/concrete.ml.torch.hybrid_model.md b/docs/references/api/concrete.ml.torch.hybrid_model.md
index da91e7bb2..96c5596b8 100644
--- a/docs/references/api/concrete.ml.torch.hybrid_model.md
+++ b/docs/references/api/concrete.ml.torch.hybrid_model.md
@@ -12,7 +12,7 @@ Implement the conversion of a torch model to a hybrid fhe/torch inference.
______________________________________________________________________
-
+
## function `tuple_to_underscore_str`
@@ -32,7 +32,7 @@ Convert a tuple to a string representation.
______________________________________________________________________
-
+
## function `underscore_str_to_tuple`
@@ -52,7 +52,7 @@ Convert a a string representation of a tuple to a tuple.
______________________________________________________________________
-
+
## function `convert_conv1d_to_linear`
@@ -72,7 +72,7 @@ Convert all Conv1D layers in a module or a Conv1D layer itself to nn.Linear.
______________________________________________________________________
-
+
## class `HybridFHEMode`
@@ -80,13 +80,13 @@ Simple enum for different modes of execution of HybridModel.
______________________________________________________________________
-
+
## class `RemoteModule`
A wrapper class for the modules to be evaluated remotely with FHE.
-
+
### method `__init__`
@@ -102,7 +102,7 @@ __init__(
______________________________________________________________________
-
+
### method `forward`
@@ -133,7 +133,7 @@ To change the behavior of this forward function one must change the fhe_local_mo
______________________________________________________________________
-
+
### method `init_fhe_client`
@@ -157,7 +157,7 @@ Set the clients keys.
______________________________________________________________________
-
+
### method `remote_call`
@@ -177,7 +177,7 @@ Call the remote server to get the private module inference.
______________________________________________________________________
-
+
## class `HybridFHEModel`
@@ -193,7 +193,7 @@ This is done by converting targeted modules by RemoteModules. This will modify t
- `model_name` (str): Model name identifier
- `verbose` (int): If logs should be printed when interacting with FHE server
-
+
### method `__init__`
@@ -209,7 +209,7 @@ __init__(
______________________________________________________________________
-
+
### method `compile_model`
@@ -235,7 +235,7 @@ Compiles the specific layers to FHE.
______________________________________________________________________
-
+
### method `init_client`
@@ -255,7 +255,7 @@ Initialize client for all remote modules.
______________________________________________________________________
-
+
### method `publish_to_hub`
@@ -267,7 +267,7 @@ Allow the user to push the model and FHE required files to HF Hub.
______________________________________________________________________
-
+
### method `save_and_clear_private_info`
@@ -284,7 +284,7 @@ Save the PyTorch model to the provided path and also saves the corresponding FHE
______________________________________________________________________
-
+
### method `set_fhe_mode`
@@ -300,7 +300,7 @@ Set Hybrid FHE mode for all remote modules.
______________________________________________________________________
-
+
## class `LoggerStub`
@@ -308,7 +308,7 @@ Placeholder type for a typical logger like the one from loguru.
______________________________________________________________________
-
+
### method `info`
@@ -324,7 +324,7 @@ Placholder function for logger.info.
______________________________________________________________________
-
+
## class `HybridFHEModelServer`
@@ -332,7 +332,7 @@ Hybrid FHE Model Server.
This is a class object to server FHE models serialized using HybridFHEModel.
-
+
### method `__init__`
@@ -342,7 +342,7 @@ __init__(key_path: Path, model_dir: Path, logger: Optional[LoggerStub])
______________________________________________________________________
-
+
### method `add_key`
@@ -365,7 +365,7 @@ Dict\[str, str\]
______________________________________________________________________
-
+
### method `check_inputs`
@@ -391,7 +391,7 @@ Check that the given configuration exist in the compiled models folder.
______________________________________________________________________
-
+
### method `compute`
@@ -421,7 +421,7 @@ Compute the circuit over encrypted input.
______________________________________________________________________
-
+
### method `dump_key`
@@ -438,7 +438,7 @@ Dump a public key to a stream.
______________________________________________________________________
-
+
### method `get_circuit`
@@ -460,7 +460,7 @@ Get circuit based on model name, module name and input shape.
______________________________________________________________________
-
+
### method `get_client`
@@ -486,7 +486,7 @@ Get client.
______________________________________________________________________
-
+
### method `list_modules`
@@ -505,7 +505,7 @@ Dict\[str, Dict\[str, Dict\]\]
______________________________________________________________________
-
+
### method `list_shapes`
@@ -525,7 +525,7 @@ Dict\[str, Dict\]
______________________________________________________________________
-
+
### method `load_key`
diff --git a/pyproject.toml b/pyproject.toml
index 09556f4db..da336722d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
[tool.poetry]
name = "concrete-ml"
-version = "1.6.0"
+version = "1.6.1"
description = "Concrete ML is an open-source set of tools which aims to simplify the use of fully homomorphic encryption (FHE) for data scientists."
license = "BSD-3-Clause-Clear"
authors = [
diff --git a/src/concrete/ml/version.py b/src/concrete/ml/version.py
index 3e9c1ae30..9a243dbf9 100644
--- a/src/concrete/ml/version.py
+++ b/src/concrete/ml/version.py
@@ -1,4 +1,4 @@
"""File to manage the version of the package."""
# Auto-generated by "make set_version" do not modify
-__version__ = "1.6.0"
+__version__ = "1.6.1"