Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,7 @@ Get text embeddings in just a few lines. The tokenizer and model are designed to
|
|
| 24 |
|
| 25 |
```python
|
| 26 |
import os
|
|
|
|
| 27 |
# Set environment for JIT compilation (Optional, set to '1' for CUDA acceleration)
|
| 28 |
os.environ["RWKV_CUDA_ON"] = '1'
|
| 29 |
|
|
|
|
| 24 |
|
| 25 |
```python
|
| 26 |
import os
|
| 27 |
+
from torch.nn import functional as F
|
| 28 |
# Set environment for JIT compilation (Optional, set to '1' for CUDA acceleration)
|
| 29 |
os.environ["RWKV_CUDA_ON"] = '1'
|
| 30 |
|