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 |
|