eyad-silx commited on
Commit
038bacd
·
verified ·
1 Parent(s): 0422df2

Update configuration_quasrav4.py

Browse files
Files changed (1) hide show
  1. configuration_quasrav4.py +4 -4
configuration_quasrav4.py CHANGED
@@ -1,11 +1,11 @@
1
  from transformers import PretrainedConfig
2
 
3
- class QuasraV4Config(PretrainedConfig):
4
  r"""
5
- This is the configuration class to store the configuration of a [`QuasraV4Model`]. It is used to instantiate a
6
- QuasraV4 model according to the specified arguments, defining the model architecture.
7
  """
8
- model_type = "quasarv4"
9
 
10
  def __init__(self, **kwargs):
11
  # Pop custom arguments from kwargs, using defaults from your config.json
 
1
  from transformers import PretrainedConfig
2
 
3
+ class InfinityFormerConfig(PretrainedConfig):
4
  r"""
5
+ This is the configuration class to store the configuration of a [`InfinityFormerModel`]. It is used to instantiate an
6
+ InfinityFormer model according to the specified arguments, defining the model architecture.
7
  """
8
+ model_type = "infinity_former"
9
 
10
  def __init__(self, **kwargs):
11
  # Pop custom arguments from kwargs, using defaults from your config.json