Skip to content

Commit 9fe754c

Browse files
committed
update test_dropout
1 parent 592d99b commit 9fe754c

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

README.md

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,21 @@ Tried Training a simple MLP with the TritonDropout, looks like its doing good.
5252
- As for the softmax I actually tests on `causal=False`
5353

5454

55-
You can run the tests individually
56-
before running the tests make sure to install triformer using
55+
To run the tests
5756

58-
`pip install -U triformer`
57+
- First git clone the repo
58+
```bash
59+
git clone https://github.com/dame-cell/Triformer.git
60+
```
61+
- Then navigate to the Triformer/tests directory
62+
```bash
63+
cd Triformer/tests
64+
```
65+
- Install triformer
66+
```bash
67+
pip install -U triformer
68+
```
69+
- Then run the tests
5970
```bash
6071
pytest tests/test_layernorm.py
6172
pytest tests/test_softmax.py

0 commit comments

Comments
 (0)