Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Make breakpoint a ContextModule

Browse source
This commit is contained in:
limiteinductive 2023-08-31 12:12:12 +02:00 committed by Benjamin Trom
parent 5327d894d1
commit ebfa51f662
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/fluxion/layers/chain.py
View file

@ -452,7 +452,7 @@ class Residual(Sum):
super().__init__(Identity(), Chain(*modules)) super().__init__(Identity(), Chain(*modules))
class Breakpoint(Module): class Breakpoint(ContextModule):
def __init__(self, vscode: bool = True): def __init__(self, vscode: bool = True):
super().__init__() super().__init__()
self.vscode = vscode self.vscode = vscode