Issues with CodeFormer and nightly version of PyTorch 1.14.0.dev2022110

[Lastek]

I'm having an issue running codeformer using nightly build of pytorch 'pytorch 1.14.0.dev20221107 py3.10_cuda11.7_cudnn8_0 pytorch-nightly'

It seems that it's not able to parse the version number correctly:

  File "F:\Dev\stable-diffusion-webui\modules\codeformer_model.py", line 38, in setup_model
    from facelib.utils.face_restoration_helper import FaceRestoreHelper
  File "F:\Dev\stable-diffusion-webui\repositories\CodeFormer\facelib\utils\face_restoration_helper.py", line 7, in <module>
    from facelib.detection import init_detection_model
  File "F:\Dev\stable-diffusion-webui\repositories\CodeFormer\facelib\detection\__init__.py", line 11, in <module>
    from .yolov5face.face_detector import YoloDetector
  File "F:\Dev\stable-diffusion-webui\repositories\CodeFormer\facelib\detection\yolov5face\face_detector.py", line 20, in <module>
    IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.'))) >= (1, 9, 0)
ValueError: invalid literal for int() with base 10: 'dev20221107'```

Python v3.10.6
Windows 10

[sczhou]

You need to pull the repo and update your local code, this issue had been fixed.

For the file: CodeFormer\facelib\detection\yolov5face\face_detector.py
IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.'))) >= (1, 9, 0)
was changed to:
IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:2])) >= (1, 9, 0)