ModuleNotFoundError: No module named 'lavis.models.blip_diffusion_models'

[f-amerehi]

Hi, Thanks for the repository and codes. I'd like to run the stylization notebook from here. When calling it, I receive the following error:

import torch

import numpy as np

from PIL import Image
from lavis.models import load_model_and_preprocess
from lavis.models.blip_diffusion_models.utils import preprocess_canny

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-3-6094afef0f8f>](https://localhost:8080/#) in <cell line: 7>()
      5 from PIL import Image
      6 from lavis.models import load_model_and_preprocess
----> 7 from lavis.models.blip_diffusion_models.utils import preprocess_canny

ModuleNotFoundError: No module named 'lavis.models.blip_diffusion_models'

---------------------------------------------------------------------------

I already pip installed !pip install salesforce-lavis. And there's indeed the utils file. Is there anything else that I need to install separately?

Thanks again.