Update modeling_btlm.py
Browse files- modeling_btlm.py +2 -2
modeling_btlm.py
CHANGED
|
@@ -43,8 +43,8 @@ from transformers.utils import (
|
|
| 43 |
logging,
|
| 44 |
)
|
| 45 |
from transformers.utils.model_parallel_utils import assert_device_map, get_device_map
|
| 46 |
-
|
| 47 |
-
from configuration_btlm import BTLMConfig
|
| 48 |
|
| 49 |
|
| 50 |
logger = logging.get_logger(__name__)
|
|
|
|
| 43 |
logging,
|
| 44 |
)
|
| 45 |
from transformers.utils.model_parallel_utils import assert_device_map, get_device_map
|
| 46 |
+
from .configuration_btlm import BTLMConfig
|
| 47 |
+
# from configuration_btlm import BTLMConfig
|
| 48 |
|
| 49 |
|
| 50 |
logger = logging.get_logger(__name__)
|