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,[15] as well as 5 games at the 2015 African U-20 Championship.[16][17][18][19] He also appeared for the Mali U23s at the 2016 Toulon Tournament.[20]