From 1ad4e1a35a70a525374f48e5abb24ba8ef6fbba6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Mon, 21 Aug 2023 15:51:17 +0200 Subject: [PATCH] converter: add missing structural_attrs --- src/refiners/fluxion/layers/converter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/refiners/fluxion/layers/converter.py b/src/refiners/fluxion/layers/converter.py index f1e7d18..653b3a8 100644 --- a/src/refiners/fluxion/layers/converter.py +++ b/src/refiners/fluxion/layers/converter.py @@ -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