eyad-silx commited on
Commit
da44902
·
verified ·
1 Parent(s): d35e524

Update configuration_quasrav4.py

Browse files
Files changed (1) hide show
  1. configuration_quasrav4.py +1 -0
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`.