From dd29409a68b3943a0eaa55b7f8e53a63577a0417 Mon Sep 17 00:00:00 2001
From: Archana Ramalingam
 <98564406+archana-ramalingam@users.noreply.github.com>
Date: Thu, 5 Sep 2024 16:18:39 -0700
Subject: [PATCH] Fix rotary_embedding.py

---
 sharktank/sharktank/layers/rotary_embedding.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/sharktank/sharktank/layers/rotary_embedding.py b/sharktank/sharktank/layers/rotary_embedding.py
index 705423aff..eadd9e6b0 100644
--- 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,