Update modeling_minGRULM.py
Browse files- modeling_minGRULM.py +1 -0
modeling_minGRULM.py
CHANGED
|
@@ -4,6 +4,7 @@ from transformers import PreTrainedModel, AutoModelForCausalLM, AutoTokenizer
|
|
| 4 |
from transformers.modeling_outputs import CausalLMOutputWithPast
|
| 5 |
from torch.nn import CrossEntropyLoss
|
| 6 |
from typing import Optional
|
|
|
|
| 7 |
from .configuration_minGRULM import MinGRULMConfig
|
| 8 |
from minGRU_pytorch.minGRULM import minGRULM
|
| 9 |
|
|
|
|
| 4 |
from transformers.modeling_outputs import CausalLMOutputWithPast
|
| 5 |
from torch.nn import CrossEntropyLoss
|
| 6 |
from typing import Optional
|
| 7 |
+
import os
|
| 8 |
from .configuration_minGRULM import MinGRULMConfig
|
| 9 |
from minGRU_pytorch.minGRULM import minGRULM
|
| 10 |
|