Skip to content

Commit

Permalink
reverse previous changes for separate PR
Browse files Browse the repository at this point in the history
  • Loading branch information
yibozhong committed Jan 19, 2025
1 parent 6deb624 commit e1b05da
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions fla/layers/abc.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,6 @@ def __init__(
use_input_gate: bool = False,
use_output_gate: bool = True,
use_norm: bool = True,
use_rope: bool = False, # FIXME
clamp_min: Optional[float] = -32,
clamp_max: Optional[float] = 32,
layer_idx: Optional[int] = None,
Expand All @@ -65,7 +64,6 @@ def __init__(
self.use_input_gate = use_input_gate
self.use_output_gate = use_output_gate
self.use_norm = use_norm
self.use_rope = use_rope # FIXME

if num_slots is None:
num_slots = self.head_k_dim
Expand Down

0 comments on commit e1b05da

Please sign in to comment.