Update test_sageattn3.py
Browse files- test_sageattn3.py +3 -1
test_sageattn3.py
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
-
# Based on
|
|
|
|
|
|
|
| 3 |
|
| 4 |
import torch
|
| 5 |
import torch.nn.functional as F
|
|
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
+
# Based on
|
| 3 |
+
# https://raw.githubusercontent.com/woct0rdho/SageAttention/refs/heads/main/tests/test_sageattn.py
|
| 4 |
+
# https://huggingface.co/jt-zhang/SageAttention3
|
| 5 |
|
| 6 |
import torch
|
| 7 |
import torch.nn.functional as F
|