mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +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.
|
||||
"""
|
||||
|
||||
structural_attrs = ["set_device", "set_dtype"]
|
||||
|
||||
def __init__(self, set_device: bool = True, set_dtype: bool = True) -> None:
|
||||
super().__init__()
|
||||
self.set_device = set_device
|
||||
|
|
Loading…
Reference in a new issue