Update configuration_quasrav4.py
Browse files
configuration_quasrav4.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
from transformers import PretrainedConfig
|
2 |
|
3 |
-
class
|
4 |
r"""
|
5 |
-
This is the configuration class to store the configuration of a [`
|
6 |
-
|
7 |
"""
|
8 |
-
model_type = "
|
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
|