脚本 inference_pipeline.ipynb 无法运行! ImportError: 无法导入名为 'partitioning' 的模块

_libgcc_mutex 0.1 main
_openmp_mutex 5.1 1_gnu
absl-py 1.4.0 pypi_0 pypi appdirs 1.4.4 pypi_0 pypi backcall 0.2.0 pypi_0 pypi ca-certificates 2023.05.30 h06a4308_0
certifi 2021.5.30 py36h06a4308_0
charset-normalizer 2.0.12 pypi_0 pypi chex 0.0.9 pypi_0 pypi click 8.0.4 pypi_0 pypi cycler 0.11.0 pypi_0 pypi dalle-mini 0.1.1 pypi_0 pypi dataclasses 0.8 pypi_0 pypi decorator 5.1.1 pypi_0 pypi dm-tree 0.1.8 pypi_0 pypi Docker-pycreds 0.4.0 pypi_0 pypi einops 0.4.1 pypi_0 pypi emoji 2.7.0 pypi_0 pypi entrypoints 0.4 pypi_0 pypi filelock 3.4.1 pypi_0 pypi flatbuffers 23.5.26 pypi_0 pypi flax 0.3.5 pypi_0 pypi ftfy 6.0.3 pypi_0 pypi gitdb 4.0.9 pypi_0 pypi gitpython 3.1.18 pypi_0 pypi huggingface-hub 0.4.0 pypi_0 pypi idna 3.4 pypi_0 pypi importlib-metadata 4.8.3 pypi_0 pypi importlib-resources 5.4.0 pypi_0 pypi ipykernel 5.5.6 pypi_0 pypi ipython 7.16.3 pypi_0 pypi ipython-genutils 0.2.0