mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-29 22:12:07 +00:00
7cdbd694b3
wip(gaudi): fix typos wip(gaudi): refactor version numbers for pytorch and habana software to make it more flexible wip(gaudi): debugging the refactored server wip(gaudi): delete useless files fix(gaudi): server working after refactoring fix(gaudi): refactor and implement requested changes
13 lines
348 B
Python
13 lines
348 B
Python
from optimum.habana.utils import get_driver_version
|
|
from packaging.version import Version
|
|
|
|
MIN_TGI_GAUDI_SYNAPSE_VERSION = Version("1.19.0")
|
|
|
|
|
|
def is_driver_compatible():
|
|
driver_version = get_driver_version()
|
|
if driver_version is not None:
|
|
if driver_version < MIN_TGI_GAUDI_SYNAPSE_VERSION:
|
|
return False
|
|
return True
|