Skip to content

Commit d3d64c6

Browse files
Revert "Add pinned numpy and fix build (pytorch#155129)"
This reverts commit a3098a7. Reverted pytorch#155129 on behalf of https://github.com/malfet due to Broke test_spectral_op, looks like missing xfail, see https://hud.pytorch.org/hud/pytorch/pytorch/0db3e0cf29604dae1007a678603e4dffd1c57562/1?per_page=50&name_filter=linux-jammy-py3.9-gcc11&mergeEphemeralLF=true ([comment](pytorch#155129 (comment)))
1 parent 0db3e0c commit d3d64c6

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

.ci/docker/requirements-docs.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,5 +58,3 @@ sphinx-copybutton==0.5.0
5858
sphinx-design==0.4.0
5959
sphinxcontrib-mermaid==1.0.0
6060
myst-parser==0.18.1
61-
62-
numpy==1.26

torch/nn/attention/bias.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,14 @@ class CausalVariant(IntEnum):
3636
3737
Defines two types of causal biases:
3838
39-
``UPPER_LEFT``: Represents upper-left triangular bias for standard causal attention.
39+
`UPPER_LEFT`: Represents upper-left triangular bias for standard causal attention.
4040
The equivalent pytorch code for constructing this bias is:
4141
4242
.. code-block:: python
4343
4444
torch.tril(torch.ones(size, dtype=torch.bool))
4545
46-
For instance, with ``shape=(3,4)``, the materialized bias tensor will be:
46+
For instance, with `shape=(3,4)`, the materialized bias tensor will be:
4747
4848
.. code-block:: text
4949
@@ -52,7 +52,7 @@ class CausalVariant(IntEnum):
5252
[1, 1, 1, 0]]
5353
5454
55-
``LOWER_RIGHT``: Represents lower-right triangular bias, the include values are aligned to the lower
55+
`LOWER_RIGHT`: Represents lower-right triangular bias, the include values are aligned to the lower
5656
right corner of the matrix.
5757
5858
The equivalent pytorch code for constructing this bias is:
@@ -65,7 +65,7 @@ class CausalVariant(IntEnum):
6565
diagonal=diagonal_offset,
6666
)
6767
68-
For instance, with ``shape=(3,4)``, the materialized bias tensor will be:
68+
For instance, with `shape=(3,4)`, the materialized bias tensor will be:
6969
7070
.. code-block:: text
7171

0 commit comments

Comments
 (0)