mirror of
https://github.com/huggingface/kernels.git
synced 2025-10-20 12:33:46 +08:00
Fix typo in layers documentation (#116)
This commit is contained in:
@ -107,7 +107,7 @@ model = kernelize(model, device="cuda", mode=Mode.INFERENCE)
|
||||
|
||||
If the `TRAINING` and/or `TORCH_COMPILE` modes are used, but a registered
|
||||
kernel does not support backward passes or `torch.compile` respectively,
|
||||
`kernenize` will fall back to the original, non-kernelized, layer. You
|
||||
`kernelize` will fall back to the original, non-kernelized, layer. You
|
||||
can let `kernelize` raise an exception instead by using `use_fallback=False`:
|
||||
|
||||
```python
|
||||
|
Reference in New Issue
Block a user