search

sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
114 stars 14 forks source link

ModuleNotFoundError: No module named 'taming.modules.vqvae' #16

Closed moxiaoli9 closed 4 months ago

moxiaoli9 commented 4 months ago

Nice work! I wonder if the vqvae.py is missing in taming.modules when I was trying to run main.py?

xuyu666 commented 4 months ago

企业微信截图_17096964247583

Replace quantize.py on your machine with this file https://github.com/CompVis/taming-transformers/blob/master/taming/modules/vqvae/quantize.py