Method init_freqs

wan/modules/multitalk_model.py:548–555

(self)

            self.init_weights()
            
    def init_freqs(self):
        d = self.dim // self.num_heads
        self.freqs = torch.cat([
            rope_params(1024, d - 4 * (d // 6)),
            rope_params(1024, 2 * (d // 6)),
            rope_params(1024, 2 * (d // 6))
        ],
                               dim=1)

    def teacache_init(
        self,