Diabaté lék einn leik með U20 liði Mali á HM U20 árið 2015 í tapinu 2<unk>0 fyrir Serbíu U20, auk þess að spila 5 leiki á
2015 Afríkur U-20 meistaraflokkur. Hann lék einnig fyrir U23 lið Mali á Toulon mótinu 2016.