Update configuration_falcon.py
Browse files- configuration_falcon.py +3 -1
configuration_falcon.py
CHANGED
@@ -15,7 +15,7 @@
|
|
15 |
""" Falcon configuration"""
|
16 |
from transformers.configuration_utils import PretrainedConfig
|
17 |
from transformers.utils import logging
|
18 |
-
|
19 |
|
20 |
logger = logging.get_logger(__name__)
|
21 |
|
@@ -25,6 +25,8 @@ FALCON_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
|
25 |
}
|
26 |
|
27 |
|
|
|
|
|
28 |
class FalconConfig(PretrainedConfig):
|
29 |
r"""
|
30 |
This is the configuration class to store the configuration of a [`FalconModel`]. It is used to instantiate a Falcon
|
|
|
15 |
""" Falcon configuration"""
|
16 |
from transformers.configuration_utils import PretrainedConfig
|
17 |
from transformers.utils import logging
|
18 |
+
from transformers.models.auto import modeling_auto
|
19 |
|
20 |
logger = logging.get_logger(__name__)
|
21 |
|
|
|
25 |
}
|
26 |
|
27 |
|
28 |
+
modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
|
29 |
+
|
30 |
class FalconConfig(PretrainedConfig):
|
31 |
r"""
|
32 |
This is the configuration class to store the configuration of a [`FalconModel`]. It is used to instantiate a Falcon
|