Maximilian Beck commited on
Commit ·
8b0f577
1
Parent(s): 044aabb
fix import
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ xlstm_config.sequence_kernel = "native_sequence__native"
|
|
| 45 |
xlstm = AutoModelForCausalLM.from_pretrained("NX-AI/xLSTM-7b", config=xlstm_config, device_map="auto")
|
| 46 |
|
| 47 |
# verify selected kernels
|
| 48 |
-
import pprint
|
| 49 |
pprint(xlstm.backbone.blocks[0].mlstm_layer.config)
|
| 50 |
```
|
| 51 |
|
|
|
|
| 45 |
xlstm = AutoModelForCausalLM.from_pretrained("NX-AI/xLSTM-7b", config=xlstm_config, device_map="auto")
|
| 46 |
|
| 47 |
# verify selected kernels
|
| 48 |
+
from pprint import pprint
|
| 49 |
pprint(xlstm.backbone.blocks[0].mlstm_layer.config)
|
| 50 |
```
|
| 51 |
|