Commit ·
f472e23
1
Parent(s): 31c63f9
Path fix
Browse files- modeling_ministu.py +0 -3
modeling_ministu.py
CHANGED
|
@@ -105,9 +105,6 @@ class MiniSTU(PreTrainedModel):
|
|
| 105 |
return CausalLMOutput(
|
| 106 |
loss=loss,
|
| 107 |
logits=logits,
|
| 108 |
-
past_key_values=None,
|
| 109 |
-
hidden_states=None,
|
| 110 |
-
attentions=None
|
| 111 |
)
|
| 112 |
|
| 113 |
def _get_num_params(self):
|
|
|
|
| 105 |
return CausalLMOutput(
|
| 106 |
loss=loss,
|
| 107 |
logits=logits,
|
|
|
|
|
|
|
|
|
|
| 108 |
)
|
| 109 |
|
| 110 |
def _get_num_params(self):
|