错误:
File "/mnt/d/Pycharm_workspace/text2image/OmniGen-version/OmniGen/latentDiffusion/ldm/models/autoencoder.py", line 6, in <module>
from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
ModuleNotFoundError: No module named 'taming.modules.vqvae'
解决办法:
pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
或者提前下载好(GitHub - CompVis/taming-transformers: Taming Transformers for High-Resolution Image Synthesis)这个先下载好。
然后cd进入该包
pip install -e .
成功