Update modeling_stablelm_epoch.py
Browse files
modeling_stablelm_epoch.py
CHANGED
@@ -27,7 +27,6 @@ import torch.utils.checkpoint
|
|
27 |
from torch import nn
|
28 |
from torch.nn import CrossEntropyLoss
|
29 |
|
30 |
-
from transformers.cache_utils import Cache
|
31 |
from transformers.modeling_outputs import (
|
32 |
BaseModelOutputWithPast,
|
33 |
CausalLMOutputWithPast,
|
|
|
27 |
from torch import nn
|
28 |
from torch.nn import CrossEntropyLoss
|
29 |
|
|
|
30 |
from transformers.modeling_outputs import (
|
31 |
BaseModelOutputWithPast,
|
32 |
CausalLMOutputWithPast,
|