Commit
·
626ae5c
1
Parent(s):
ff98655
fixes
Browse files
__pycache__/configuration_ministu.cpython-312.pyc
ADDED
|
Binary file (2.39 kB). View file
|
|
|
__pycache__/modeling_ministu.cpython-312.pyc
ADDED
|
Binary file (33 kB). View file
|
|
|
modeling_ministu.py
CHANGED
|
@@ -6,6 +6,7 @@ import torch.nn as nn
|
|
| 6 |
import torch.nn.functional as F
|
| 7 |
from transformers import PreTrainedModel, PretrainedConfig
|
| 8 |
from .configuration_ministu import MiniSTUConfig
|
|
|
|
| 9 |
try:
|
| 10 |
from flashfftconv import FlashFFTConv
|
| 11 |
|
|
|
|
| 6 |
import torch.nn.functional as F
|
| 7 |
from transformers import PreTrainedModel, PretrainedConfig
|
| 8 |
from .configuration_ministu import MiniSTUConfig
|
| 9 |
+
from transformers.modeling_outputs import CausalLMOutput
|
| 10 |
try:
|
| 11 |
from flashfftconv import FlashFFTConv
|
| 12 |
|