Searched refs:precompute_freqs_cis (Results 1 – 5 of 5) sorted by relevance
20 precompute_freqs_cis,469 self.precompute_freqs_cis = hf_precompute_freqs_cis471 self.precompute_freqs_cis = partial(472 precompute_freqs_cis, use_scaled=params.use_scaled_rope474 freqs_cos, freqs_sin = self.precompute_freqs_cis(
43 def precompute_freqs_cis( function
144 self.freqs_cis = precompute_freqs_cis(269 def precompute_freqs_cis(seq_len: int, n_elem: int, base: int = 10000) -> Tensor: function
123 self.freqs_cis = precompute_freqs_cis(282 def precompute_freqs_cis(seq_len: int, n_elem: int, base: int = 10000) -> Tensor: function
15 precompute_freqs_cis,241 freqs_cos, freqs_sin = precompute_freqs_cis(