mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-13 00:28:14 +00:00
converter: add missing structural_attrs
This commit is contained in:
parent
b91a457495
commit
1ad4e1a35a
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue