Where is frequency-aware RoPE is implemented?

Hi authors, fantastic work! I see where time-aware RoPE is implemented, but I can't seem to find the frequency-aware RoPE as specified in Sec 2.2. Could you please help with that? Thanks!