Skip to content

Commit 08a0ea6

Browse files
committed
append hidden imports
1 parent b6ebe39 commit 08a0ea6

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

iotdb-core/ainode/ainode.spec

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,24 @@ for lib in essential_libraries:
6464
except Exception:
6565
pass
6666

67+
# Some dependencies might still miss in specified operation systems, manually import them in this case
68+
extra_hidden = [
69+
# torch dynamo polyfills
70+
'torch._dynamo.polyfills',
71+
'torch._dynamo.polyfills.functools',
72+
73+
# torch flex attention
74+
'torch.nn.attention.flex_attention',
75+
76+
# transformers
77+
'transformers.masking_utils',
78+
'transformers.generation.utils',
79+
'transformers.models.auto.modeling_auto',
80+
'transformers.models.auto.auto_factory',
81+
]
82+
83+
all_hiddenimports.extend(extra_hidden)
84+
6785
# Project-specific packages that need their submodules collected
6886
# Only list top-level packages - collect_submodules will recursively collect all submodules
6987
TOP_LEVEL_PACKAGES = [

0 commit comments

Comments
 (0)