Add check for hpu and wrap_in_hpu_graph availability. #3249
Conversation
|
Thank you for the PR @vshekhawat-hlab. @ZhengHongming888 Could you have a look at this? I'm unfamiliar with the correct usage of the Habana library for HPU support.
|
vshekhawat-hlab
force-pushed
the
disable_hpu_graph_eager_mode
branch
from
f1600b9 to
309a9ac
Compare
vshekhawat-hlab
force-pushed
the
disable_hpu_graph_eager_mode
branch
from
0bc899d to
9690034
Compare
vshekhawat-hlab
changed the title
|
Added check for hpu and wrap_in_hpu_graph availability. This Prevent AttributeError by verifying that habana_frameworks.torch.hpu and its wrap_in_hpu_graph function exist. This ensures compatibility with eager mode. |
yes I am ok with this PR. Thanks @tomaarsen . |
Prevent AttributeError by verifying that habana_frameworks.torch.hpu and its wrap_in_hpu_graph function exist before calling it.
vshekhawat-hlab
force-pushed
the
disable_hpu_graph_eager_mode
branch
from
9690034 to
4ab9038
Compare
|
Can we retrigger the CI jobs, as seems HF connection errors. Is it known issue? |
|
It's a known issue indeed, the CI container sometimes gets flagged by HF as making too many requests, so then it cant load the models nicely.
|
No description provided.