dmx.compressor.modeling.nn.custom_modules.apply_rotary_embeddings

dmx.compressor.modeling.nn.custom_modules.apply_rotary_embeddings(x, cos_embedding, sin_embedding)