klemenk commited on
Commit
bb957c9
·
verified ·
1 Parent(s): 762ba42

Update modeling_auristream.py

Browse files
Files changed (1) hide show
  1. modeling_auristream.py +1 -1
modeling_auristream.py CHANGED
@@ -80,7 +80,7 @@ class AuriStream(PreTrainedModel):
80
  output_hidden_states=False,
81
  return_dict=False,
82
  up_until_layer=None,
83
- normalize_embeddings=None, # <- NEW (defaults to None)
84
  ):
85
  """
86
  Input: seq: torch.Tensor of shape (b, t)
 
80
  output_hidden_states=False,
81
  return_dict=False,
82
  up_until_layer=None,
83
+ normalize_embeddings=None,
84
  ):
85
  """
86
  Input: seq: torch.Tensor of shape (b, t)