Update modeling_cpmbee.py
Browse files- modeling_cpmbee.py +6 -6
modeling_cpmbee.py
CHANGED
@@ -21,9 +21,9 @@ from typing import Any, Callable, Dict, List, Optional, Tuple, Union
|
|
21 |
import torch
|
22 |
import torch.nn as nn
|
23 |
|
24 |
-
from
|
25 |
-
from
|
26 |
-
from
|
27 |
GenerationConfig,
|
28 |
LogitsProcessorList,
|
29 |
StoppingCriteriaList,
|
@@ -32,9 +32,9 @@ from ...generation.utils import (
|
|
32 |
is_deepspeed_zero3_enabled,
|
33 |
warnings,
|
34 |
)
|
35 |
-
from
|
36 |
-
from
|
37 |
-
from
|
38 |
from .configuration_cpmbee import CpmBeeConfig
|
39 |
from .tokenization_cpmbee import CpmBeeTokenizer
|
40 |
|
|
|
21 |
import torch
|
22 |
import torch.nn as nn
|
23 |
|
24 |
+
from transformers.generation.beam_search import BeamHypotheses, BeamSearchScorer
|
25 |
+
from transformers.generation.streamers import BaseStreamer
|
26 |
+
from transformers.generation.utils import (
|
27 |
GenerationConfig,
|
28 |
LogitsProcessorList,
|
29 |
StoppingCriteriaList,
|
|
|
32 |
is_deepspeed_zero3_enabled,
|
33 |
warnings,
|
34 |
)
|
35 |
+
from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, ModelOutput
|
36 |
+
from transformers.modeling_utils import PreTrainedModel
|
37 |
+
from transformers.utils import add_code_sample_docstrings, add_start_docstrings, add_start_docstrings_to_model_forward, logging
|
38 |
from .configuration_cpmbee import CpmBeeConfig
|
39 |
from .tokenization_cpmbee import CpmBeeTokenizer
|
40 |
|