ModuleNotFoundError: No module named 'models'

[guluahljj]

E:\ZZZ_Github\YOLO-World\demo\sample_images 'person,dog,cat' --topk 100 --threshold 0.005 --output-dir demo_outputs
Loads checkpoint by local backend from path: ./weights/yolov5s.pt
Traceback (most recent call last):
File "E:\ZZZ_Github\YOLO-World\demo\image_demo.py", line 179, in
model = init_detector(cfg, checkpoint=args.checkpoint, device=args.device)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\mmdet\apis\inference.py", line 68, in init_detector
checkpoint = load_checkpoint(model, checkpoint, map_location='cpu')
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\mmengine\runner\checkpoint.py", line 636, in load_checkpoint
checkpoint = _load_checkpoint(filename, map_location, logger)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\mmengine\runner\checkpoint.py", line 548, in _load_checkpoint
return CheckpointLoader.load_checkpoint(filename, map_location, logger)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\mmengine\runner\checkpoint.py", line 330, in load_checkpoint
return checkpoint_loader(filename, map_location)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\mmengine\runner\checkpoint.py", line 347, in load_from_local
checkpoint = torch.load(filename, map_location=map_location)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\torch\serialization.py", line 712, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\torch\serialization.py", line 1046, in _load
result = unpickler.load()
File "E:\Anaconda_Enviroments\yoloworld\lib\site-packages\torch\serialization.py", line 1039, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'models'