Skip to content

GitLab

Unverified Commit a1160185 authored by Michael Benayoun's avatar Michael Benayoun Committed by GitHub
Browse files

Fix `is_optimum_neuron_available` (#23961) 

Fix is_optimum_neuron_available
parent 6b548129
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/transformers/utils/import_utils.py
View file @ a1160185
...@@ -98,7 +98,6 @@ _natten_available = _is_package_available("natten") ...@@ -98,7 +98,6 @@ _natten_available = _is_package_available("natten")
_onnx_available = _is_package_available("onnx") _onnx_available = _is_package_available("onnx")
_openai_available = _is_package_available("openai") _openai_available = _is_package_available("openai")
_optimum_available = _is_package_available("optimum") _optimum_available = _is_package_available("optimum")
_optimumneuron_available = _optimum_available and _is_package_available("optimum.neuron")
_pandas_available = _is_package_available("pandas") _pandas_available = _is_package_available("pandas")
_peft_available = _is_package_available("peft") _peft_available = _is_package_available("peft")
_phonemizer_available = _is_package_available("phonemizer") _phonemizer_available = _is_package_available("phonemizer")
...@@ -520,7 +519,7 @@ def is_optimum_available(): ...@@ -520,7 +519,7 @@ def is_optimum_available():
def is_optimum_neuron_available(): def is_optimum_neuron_available():
return _optimumneuron_available return _optimum_available and _is_package_available("optimum.neuron")
def is_safetensors_available(): def is_safetensors_available():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment