Diabaté made one appearance for the Mali U20s at the 2015 FIFA U-20 World Cup in a 2–0 loss to the Serbia U20s, as well as 5 games at the 2015 African U-20 Championship. He also appeared for the Mali U23s at the 2016 Toulon Tournament.