Upload modeling_qwen2.py
Browse files- modeling_qwen2.py +1 -1
modeling_qwen2.py
CHANGED
|
@@ -40,7 +40,7 @@ from transformers.utils import (add_start_docstrings,
|
|
| 40 |
is_flash_attn_greater_or_equal_2_10, logging,
|
| 41 |
replace_return_docstrings)
|
| 42 |
|
| 43 |
-
from
|
| 44 |
from .configuration_qwen2 import QwenEnPRMConfig as Qwen2Config
|
| 45 |
|
| 46 |
if is_flash_attn_2_available():
|
|
|
|
| 40 |
is_flash_attn_greater_or_equal_2_10, logging,
|
| 41 |
replace_return_docstrings)
|
| 42 |
|
| 43 |
+
from .nets import EnsembleModel
|
| 44 |
from .configuration_qwen2 import QwenEnPRMConfig as Qwen2Config
|
| 45 |
|
| 46 |
if is_flash_attn_2_available():
|