YARlabs commited on
Commit
8fca974
·
verified ·
1 Parent(s): 8c29280

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- from configuration_yar import YarConfig
 
 
 
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")