We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e6eafce commit d4db9d8Copy full SHA for d4db9d8
src/transformers/models/granite_speech/processing_granite_speech.py
@@ -16,13 +16,14 @@
16
17
from typing import List, Union
18
19
-import torch
20
-
21
from transformers.feature_extraction_utils import BatchFeature
22
from transformers.processing_utils import ProcessorMixin
23
from transformers.tokenization_utils import PreTokenizedInput, TextInput
24
-from transformers.utils import logging
+from transformers.utils import is_torch_available, logging
+
25
+if is_torch_available():
26
+ import torch
27
28
logger = logging.get_logger(__name__)
29
0 commit comments