Update modeling_FalconTST.py
Browse files- modeling_FalconTST.py +2 -2
modeling_FalconTST.py
CHANGED
|
@@ -9,8 +9,8 @@ import math
|
|
| 9 |
from einops import rearrange, repeat
|
| 10 |
from functools import reduce
|
| 11 |
from abc import ABC, abstractmethod
|
| 12 |
-
from configuration_FalconTST import FalconTSTConfig
|
| 13 |
-
from ts_generation_mixin import FalconTSTGenerationMixin
|
| 14 |
from transformers import PreTrainedModel, Cache, DynamicCache
|
| 15 |
from transformers.activations import ACT2FN
|
| 16 |
from transformers.modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
|
|
|
|
| 9 |
from einops import rearrange, repeat
|
| 10 |
from functools import reduce
|
| 11 |
from abc import ABC, abstractmethod
|
| 12 |
+
from .configuration_FalconTST import FalconTSTConfig
|
| 13 |
+
from .ts_generation_mixin import FalconTSTGenerationMixin
|
| 14 |
from transformers import PreTrainedModel, Cache, DynamicCache
|
| 15 |
from transformers.activations import ACT2FN
|
| 16 |
from transformers.modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
|