Fix rotary_embedding.py

nod-ai · Sep 5, 2024 · dd29409 · dd29409
1 parent eb91f73
commit dd29409
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/sharktank/sharktank/layers/rotary_embedding.py b/sharktank/sharktank/layers/rotary_embedding.py
@@ -18,7 +18,6 @@ def __init__(
         self,
         *,
         rope_dimension_count: int,
-        rope_freq_base: float,
         max_seqlen: int,
         rope_freq_base: Optional[float] = 10000.0,
         device: Optional[torch.device] = None,