Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 00:58:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

converter: add missing structural_attrs

Browse source
This commit is contained in:
Cédric Deltheil 2023-08-21 15:51:17 +02:00 committed by Cédric Deltheil
parent b91a457495
commit 1ad4e1a35a
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -17,6 +17,8 @@ class Converter(ContextModule):
Ensure the parent module has `device` and `dtype` attributes if `set_device` or `set_dtype` are set to True. Ensure the parent module has `device` and `dtype` attributes if `set_device` or `set_dtype` are set to True.
""" """
structural_attrs = ["set_device", "set_dtype"]
def __init__(self, set_device: bool = True, set_dtype: bool = True) -> None: def __init__(self, set_device: bool = True, set_dtype: bool = True) -> None:
super().__init__() super().__init__()
self.set_device = set_device self.set_device = set_device