File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed
Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff 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
6987TOP_LEVEL_PACKAGES = [
You can’t perform that action at this time.
0 commit comments