Upload folder using huggingface_hub
Browse files- modeling_yar.py +4 -1
modeling_yar.py
CHANGED
|
@@ -42,7 +42,10 @@ from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
|
|
| 42 |
from transformers.processing_utils import Unpack
|
| 43 |
from transformers.utils import TransformersKwargs, auto_docstring, can_return_tuple
|
| 44 |
from transformers.utils.generic import check_model_inputs, maybe_autocast
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
@use_kernel_forward_from_hub("RMSNorm")
|
|
|
|
| 42 |
from transformers.processing_utils import Unpack
|
| 43 |
from transformers.utils import TransformersKwargs, auto_docstring, can_return_tuple
|
| 44 |
from transformers.utils.generic import check_model_inputs, maybe_autocast
|
| 45 |
+
try:
|
| 46 |
+
from configuration_yar import YarConfig
|
| 47 |
+
except ImportError:
|
| 48 |
+
from .configuration_yar import YarConfig
|
| 49 |
|
| 50 |
|
| 51 |
@use_kernel_forward_from_hub("RMSNorm")
|