Issue with pytorch_lightning

[dayunbao]

I just tried to run the handrefiner.py script after installing everything in the requirements.txt file, and got this error:

ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data'

I saw a lot of people have had this error in the past. The pytorch_lightning version in the requirements.txt file is pinned to 1.4.2, but the current stable version is 2.1.3. Is there a reason for using such an old version of pytorch_lightning?

[wenquanlu]

Oh, that is because when I installed the stable diffusion https://github.com/CompVis/stable-diffusion , I use their environment.yaml which is very old... I think you can use whatever version you would like as long as there is no major conflicts...