jeffreygo commited on
Commit
3cc9b7a
1 Parent(s): edc4347

Update modeling_cpmbee.py

Browse files
Files changed (1) hide show
  1. 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 ...generation.beam_search import BeamHypotheses, BeamSearchScorer
25
- from ...generation.streamers import BaseStreamer
26
- from ...generation.utils import (
27
  GenerationConfig,
28
  LogitsProcessorList,
29
  StoppingCriteriaList,
@@ -32,9 +32,9 @@ from ...generation.utils import (
32
  is_deepspeed_zero3_enabled,
33
  warnings,
34
  )
35
- from ...modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, ModelOutput
36
- from ...modeling_utils import PreTrainedModel
37
- from ...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
 
 
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