Skip to content

Commit 31e773d

Browse files
Use relative imports
1 parent d4db9d8 commit 31e773d

File tree

3 files changed

+11
-15
lines changed

3 files changed

+11
-15
lines changed

src/transformers/models/granite_speech/configuration_granite_speech.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
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

2121
class GraniteSpeechEncoderConfig(PretrainedConfig):

src/transformers/models/granite_speech/modeling_granite_speech.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,23 +19,19 @@
1919

2020
import torch
2121
import torch.nn.functional as F
22-
import torch.utils.checkpoint
2322
from 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
3935
from .configuration_granite_speech import (
4036
GraniteSpeechConfig,
4137
GraniteSpeechEncoderConfig,

src/transformers/models/granite_speech/processing_granite_speech.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@
1616

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

2525
if is_torch_available():

0 commit comments

Comments
 (0)