Update modeling_gritlm8x7b.py
Browse files- modeling_gritlm8x7b.py +1 -1
modeling_gritlm8x7b.py
CHANGED
|
@@ -54,7 +54,7 @@ from transformers.utils import (
|
|
| 54 |
replace_return_docstrings,
|
| 55 |
)
|
| 56 |
from transformers.utils.import_utils import is_torch_fx_available
|
| 57 |
-
from
|
| 58 |
|
| 59 |
# transformers has a bug where it will try to import everything from a custom model file unless there's try/except
|
| 60 |
try:
|
|
|
|
| 54 |
replace_return_docstrings,
|
| 55 |
)
|
| 56 |
from transformers.utils.import_utils import is_torch_fx_available
|
| 57 |
+
from transformers import MixtralConfig
|
| 58 |
|
| 59 |
# transformers has a bug where it will try to import everything from a custom model file unless there's try/except
|
| 60 |
try:
|