Skip to content

Commit 49d7fdb

Browse files
authored
Update README.md
1 parent 0ddfcd0 commit 49d7fdb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

example/README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ python sageattn_cogvideo.py
2323

2424
---
2525

26-
### Another Example for cogvideoX's SAT
26+
### Another Example for cogvideoX-2B SAT (35% end-to-end speedup)
2727
We will take [Cogvideo SAT](https://github.com/THUDM/CogVideo/tree/main) as an example:
2828

2929
Once you have set up the environment for cogvideoX's SAT and can generate videos, you can plug SageAttention and play easily by replacing lines 67-72 in CogVideo/sat/sat/transformer_defaults.py:
@@ -45,4 +45,4 @@ with the following code:
4545
query_layer, key_layer, value_layer,
4646
is_causal=not is_full
4747
)
48-
```
48+
```

0 commit comments

Comments
 (0)