Diabaté on edustanut Malia vuodesta 2005 lähtien. Hänet valittiin ensimmäisiin arvokisoihinsa vuoden 2012 Afrikan-mestaruuskisoihin. Hän saavutti turnauksen päätteeksi pronssia, kun Mali voitti Ghanan maalein 0–2. Diabaté oli myös Malin ja kisojen paras maalintekijä kolmella osumalla.