From ebfa51f662a90f38040a38adca0c7c58ce64ff7a Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Thu, 31 Aug 2023 12:12:12 +0200 Subject: [PATCH] Make breakpoint a ContextModule --- src/refiners/fluxion/layers/chain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/refiners/fluxion/layers/chain.py b/src/refiners/fluxion/layers/chain.py index cfabe15..6669cda 100644 --- a/src/refiners/fluxion/layers/chain.py +++ b/src/refiners/fluxion/layers/chain.py @@ -452,7 +452,7 @@ class Residual(Sum): super().__init__(Identity(), Chain(*modules)) -class Breakpoint(Module): +class Breakpoint(ContextModule): def __init__(self, vscode: bool = True): super().__init__() self.vscode = vscode