Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
ByronHsu authored Aug 15, 2024
1 parent 10ea664 commit d5609fd
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -84,20 +84,15 @@ $ pip install liger-kernel-nightly
Using [patching APIs](#patching), you can swap Hugging Face models with optimized Liger Kernels.

```python
import transformers
from liger_kernel.transformers import apply_liger_kernel_to_llama
from transformers import Trainer


model = transformers.AutoModelForCausalLM.from_pretrained("<some llama model>")

# By adding this line, it automatically monkey patches the model with the optimized kernels
apply_liger_kernel_to_llama()
```





### 2. Compose Your Own Model

You can take individual [kernels](#kernels) to compose your models.
Expand Down

0 comments on commit d5609fd

Please sign in to comment.