run demo with weights from yolov5

[chyphen7]

I am trying to run demo with a model trained on yolov5, and getting an error: No module named 'models.yolo'. Is TrTr compatible with yolo weights? Do I need convert yolo weights (*.pt) to a format compatible with TrTr, and how? Thanks,