Method __init__

wan/utils/multitalk_utils.py:156–161

(self,
                 head_dim,
                 )

class RotaryPositionalEmbedding1D(nn.Module):

    def __init__(self,
                 head_dim,
                 ):
        super().__init__()
        self.head_dim = head_dim
        self.base = 10000


    @lru_cache(maxsize=32)