Id map corner borders
Ok I finally figured it out. It is not as good as the color map from above, but its close enough... Here is what I did, if anyone has the same problem:
Add those nodes in the blender shader nodes and click on viewport shading (with cycles renderer) You can adjust the border from the bevel node.