Maxtimer97
/

Llama2SWA256

Text Generation

Model card Files Files and versions

Maxtimer97 commited on Jul 2

Commit

e817d17

·

verified ·

1 Parent(s): e2967d0

Generalized config import

Files changed (1) hide show

modeling_hymba.py +4 -2

modeling_hymba.py CHANGED Viewed

@@ -39,8 +39,10 @@ from transformers.utils import (
 from transformers.utils.import_utils import is_torch_fx_available
-from configuration_hymba import HymbaConfig
-# from .configuration_hymba import HymbaConfig
 from torch.utils.checkpoint import checkpoint

 from transformers.utils.import_utils import is_torch_fx_available
+try:
+    from configuration_hymba import HymbaConfig  # Absolute import
+except ImportError:
+    from .configuration_hymba import HymbaConfig
 from torch.utils.checkpoint import checkpoint