diff --git a/src/concrete/ml/torch/hybrid_model.py b/src/concrete/ml/torch/hybrid_model.py index 384e63c21..5fee3fee6 100644 --- a/src/concrete/ml/torch/hybrid_model.py +++ b/src/concrete/ml/torch/hybrid_model.py @@ -555,7 +555,7 @@ def save_and_clear_private_info(self, path: Path, via_mlir=True): model_path = path / "model.pth" # Save the model state dict due to a Brevitas issue - # FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/4571 + # FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/4572 torch.save(self.model.state_dict(), model_path.resolve()) # Save the FHE circuit in the same directory