all thing
Browse files
myolmoe/modeling_myolmoe.py
CHANGED
|
@@ -955,3 +955,5 @@ class MyOlmoeForCausalLM(OlmoePreTrainedModel, GenerationMixin):
|
|
| 955 |
attentions=outputs.attentions,
|
| 956 |
router_logits=outputs.router_logits,
|
| 957 |
)
|
|
|
|
|
|
|
|
|
| 955 |
attentions=outputs.attentions,
|
| 956 |
router_logits=outputs.router_logits,
|
| 957 |
)
|
| 958 |
+
|
| 959 |
+
__all__ = ["MyOlmoeForCausalLM", "OlmoeModel", "OlmoePreTrainedModel"]
|