Diabaté esiintyi kerran Malin U20-joukkueessa FIFA U-20 -maailmancupissa 2015, jossa hän hävisi Serbia U20-joukkueelle 2: 0, ja pelasi viisi peliä 2015: n Afrikan U-20-mestaruuskilpailussa.
- Mitä? Hän esiintyi myös Malin U23-joukkueessa Toulon-turnauksessa vuonna 2016.