Diabaté spillede én kamp for Mali U20 i 2015 FIFA U-20 World Cup i et 2-0 tab til Serbien U20, samt 5 kampe ved 2015 African U-20 Championship.
- Hvad? Han optrådte også for Mali U23 i Toulon-turneringen i 2016.