classname
Browse files
myolmoe/modeling_myolmoe.py
CHANGED
|
@@ -841,7 +841,7 @@ class OlmoeModel(OlmoePreTrainedModel):
|
|
| 841 |
return causal_mask
|
| 842 |
|
| 843 |
|
| 844 |
-
class
|
| 845 |
_tied_weights_keys = ["lm_head.weight"]
|
| 846 |
|
| 847 |
def __init__(self, config):
|
|
|
|
| 841 |
return causal_mask
|
| 842 |
|
| 843 |
|
| 844 |
+
class MyOlmoeForCausalLM(OlmoePreTrainedModel, GenerationMixin):
|
| 845 |
_tied_weights_keys = ["lm_head.weight"]
|
| 846 |
|
| 847 |
def __init__(self, config):
|