Correct config for IA3

Signed-off-by: smajumdar <[email protected]>
NVIDIA · Oct 15, 2022 · c89c159 · c89c159
1 parent 3b4654b
commit c89c159
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/modules/common/megatron/parallel_adapters.py b/nemo/collections/nlp/modules/common/megatron/parallel_adapters.py
@@ -55,7 +55,7 @@ def forward(self, x):
 @dataclass
 class InfusedAdapterConfig:
     in_features: int
-    adapter_strategy: Optional[dict] = adapter_mixin_strategies.ResidualAddAdapterStrategyConfig()
+    adapter_strategy: Optional[Any] = adapter_mixin_strategies.ResidualAddAdapterStrategyConfig()
     _target_: str = "{0}.{1}".format(InfusedAdapter.__module__, InfusedAdapter.__name__)