Update configuration_quasrav4.py
Browse files
configuration_quasrav4.py
CHANGED
@@ -29,6 +29,7 @@ class InfinityFormerConfig(PretrainedConfig):
|
|
29 |
self.kernel_epsilon = kwargs.pop("kernel_epsilon", 0.1)
|
30 |
self.use_gating = kwargs.pop("use_gating", True)
|
31 |
self.gate_init_bias = kwargs.pop("gate_init_bias", -2.0)
|
|
|
32 |
self.use_gradient_checkpointing = kwargs.pop("use_gradient_checkpointing", False)
|
33 |
|
34 |
# The `use_return_dict` is a read-only property that depends on `return_dict`.
|
|
|
29 |
self.kernel_epsilon = kwargs.pop("kernel_epsilon", 0.1)
|
30 |
self.use_gating = kwargs.pop("use_gating", True)
|
31 |
self.gate_init_bias = kwargs.pop("gate_init_bias", -2.0)
|
32 |
+
self.use_memory_attention = kwargs.pop("use_memory_attention", False)
|
33 |
self.use_gradient_checkpointing = kwargs.pop("use_gradient_checkpointing", False)
|
34 |
|
35 |
# The `use_return_dict` is a read-only property that depends on `return_dict`.
|