WEO-SAS
/

mamba-ssm

RhodWeo commited on 4 days ago

Commit

60ecc7d

verified ·

1 Parent(s): a7c0c91

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ This is the repository card of kernels-community/mamba-ssm that has been pushed
 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
-kernel_module = get_kernel("kernels-community/mamba-ssm")
 selective_scan_fn = kernel_module.selective_scan_fn
 selective_scan_fn(...)

 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
+kernel_module = get_kernel("WEO-SAS/mamba-ssm")
 selective_scan_fn = kernel_module.selective_scan_fn
 selective_scan_fn(...)