StableAudioDiTModel  
The Diffusion Transformer model introduced in Stable Audio.  
Reference: https://github.com/Stability-AI/stable-audio-tools  
Parameters:
sample_size ( `int`, *optional*, defaults to 1024): The size of the input sample.
in_channels (`int`, *optional*, defaults to 64): The number of channels in the input.
num_layers (`int`, *optional*, defaults to 24): The number of layers of Transformer blocks to use.