Update modeling_auristream.py
Browse files- 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,
|
| 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)
|