Update README.md
Browse files
README.md
CHANGED
|
@@ -37,7 +37,7 @@ with attention_backend("_flash_3_hub"): # or `"flash_hub"` if you are not on H10
|
|
| 37 |
|
| 38 |
To run inference with default attention backend
|
| 39 |
|
| 40 |
-
|
| 41 |
```py
|
| 42 |
import torch
|
| 43 |
|
|
|
|
| 37 |
|
| 38 |
To run inference with default attention backend
|
| 39 |
|
| 40 |
+
|
| 41 |
```py
|
| 42 |
import torch
|
| 43 |
|