OxxoCodes
/

jamba-small-v2

Text Generation

Inference Endpoints

Model card Files Files and versions Community

OxxoCodes commited on Apr 9

Commit

e9a4f70

•

1 Parent(s): 481a13a

Fix modeling_jamba.py

Files changed (1) hide show

modeling_jamba.py +1 -1

modeling_jamba.py CHANGED Viewed

@@ -51,7 +51,7 @@ from transformers.utils import (
     replace_return_docstrings,
 )
 from transformers.utils.import_utils import is_torch_fx_available
-from configuration_jamba import JambaConfig
 # try except block so it'll work with trust_remote_code. Later we can have `if is_flash_attn_2_available():`

     replace_return_docstrings,
 )
 from transformers.utils.import_utils import is_torch_fx_available
+from .configuration_jamba import JambaConfig
 # try except block so it'll work with trust_remote_code. Later we can have `if is_flash_attn_2_available():`