File tree Expand file tree Collapse file tree 3 files changed +11
-15
lines changed
src/transformers/models/granite_speech Expand file tree Collapse file tree 3 files changed +11
-15
lines changed Original file line number Diff line number Diff line change 1414# limitations under the License.
1515"""Config class for Granite Speech."""
1616
17- from transformers .configuration_utils import PretrainedConfig
18- from transformers . models .auto import CONFIG_MAPPING , AutoConfig
17+ from .. .configuration_utils import PretrainedConfig
18+ from . .auto import CONFIG_MAPPING , AutoConfig
1919
2020
2121class GraniteSpeechEncoderConfig (PretrainedConfig ):
Original file line number Diff line number Diff line change 1919
2020import torch
2121import torch .nn .functional as F
22- import torch .utils .checkpoint
2322from torch import nn
2423
25- from transformers .generation import GenerationMixin
26- from transformers .modeling_outputs import (
27- ModelOutput ,
28- )
29- from transformers .modeling_utils import PreTrainedModel
30- from transformers .models .auto import AutoModel , AutoModelForCausalLM
31- from transformers .utils import (
24+ from ...generation import GenerationMixin
25+ from ...modeling_outputs import ModelOutput
26+ from ...modeling_utils import PreTrainedModel
27+ from ...utils import (
3228 add_start_docstrings ,
3329 add_start_docstrings_to_model_forward ,
3430 is_peft_available ,
3531 logging ,
3632 replace_return_docstrings ,
3733)
38-
34+ from .. auto import AutoModel , AutoModelForCausalLM
3935from .configuration_granite_speech import (
4036 GraniteSpeechConfig ,
4137 GraniteSpeechEncoderConfig ,
Original file line number Diff line number Diff line change 1616
1717from typing import List , Union
1818
19- from transformers .feature_extraction_utils import BatchFeature
20- from transformers .processing_utils import ProcessorMixin
21- from transformers .tokenization_utils import PreTokenizedInput , TextInput
22- from transformers .utils import is_torch_available , logging
19+ from .. .feature_extraction_utils import BatchFeature
20+ from .. .processing_utils import ProcessorMixin
21+ from .. .tokenization_utils import PreTokenizedInput , TextInput
22+ from .. .utils import is_torch_available , logging
2323
2424
2525if is_torch_available ():
You can’t perform that action at this time.
0 commit comments