We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bc1f8b5 commit a033482Copy full SHA for a033482
1 file changed
sdks/python/apache_beam/ml/inference/huggingface_inference.py
@@ -32,11 +32,7 @@
32
import torch
33
from transformers import AutoModel
34
from transformers import Pipeline
35
-
36
-try:
37
- from transformers import TFAutoModel
38
-except ImportError:
39
- TFAutoModel = Any
+from transformers import TFAutoModel
40
from transformers import pipeline
41
42
from apache_beam.ml.inference import utils
0 commit comments