This isn't exactly what you asked for, but there's a "drumsep" model, which takes a drum audio track and separates it into 6 stems: kick, snare, toms, hi-hat, ride, and crash.
Ctrl+F for "drumsep" in this doc:
Instrumental, vocal & other stems separation & mix/master guide - UVR/MDX/Demucs/GSEP & others - Google Docs https://docs.google.com/document/d/17fjNvJzj8ZGSer7c7OFe_CNf...