修改readme

25991f98 · hepj · ac192496 · 25991f98 · 25991f98 · 25991f98
Commit 25991f98 authored Jul 25, 2024 by hepj
20 changed files
--- a/mamba_ssm/models/__init__.py
+++ b/mamba_ssm/models/__init__.py
--- a/mamba_ssm/models/config_mamba.py
+++ b/mamba_ssm/models/config_mamba.py
--- a/mamba_ssm/models/mixer_seq_simple.py
+++ b/mamba_ssm/models/mixer_seq_simple.py
--- a/mamba_ssm/modules/__init__.py
+++ b/mamba_ssm/modules/__init__.py
--- a/mamba_ssm/modules/block.py
+++ b/mamba_ssm/modules/block.py
--- a/mamba_ssm/modules/mamba2.py
+++ b/mamba_ssm/modules/mamba2.py
--- a/mamba_ssm/modules/mamba2_simple.py
+++ b/mamba_ssm/modules/mamba2_simple.py
--- a/mamba_ssm/modules/mamba_simple.py
+++ b/mamba_ssm/modules/mamba_simple.py
--- a/mamba_ssm/modules/mha.py
+++ b/mamba_ssm/modules/mha.py
--- a/mamba_ssm/modules/mlp.py
+++ b/mamba_ssm/modules/mlp.py
--- a/mamba_ssm/modules/ssd_minimal.py
+++ b/mamba_ssm/modules/ssd_minimal.py
--- a/mamba_ssm/ops/__init__.py
+++ b/mamba_ssm/ops/__init__.py
--- a/mamba_ssm/ops/selective_scan_interface.py
+++ b/mamba_ssm/ops/selective_scan_interface.py
--- a/mamba_ssm/ops/triton/__init__.py
+++ b/mamba_ssm/ops/triton/__init__.py
--- a/mamba_ssm/ops/triton/k_activations.py
+++ b/mamba_ssm/ops/triton/k_activations.py
--- a/mamba_ssm/ops/triton/layer_norm.py
+++ b/mamba_ssm/ops/triton/layer_norm.py
--- a/mamba_ssm/ops/triton/layernorm_gated.py
+++ b/mamba_ssm/ops/triton/layernorm_gated.py
--- a/mamba_ssm/ops/triton/selective_state_update.py
+++ b/mamba_ssm/ops/triton/selective_state_update.py
--- a/mamba_ssm/ops/triton/softplus.py
+++ b/mamba_ssm/ops/triton/softplus.py
--- a/mamba_ssm/ops/triton/ssd_bmm.py
+++ b/mamba_ssm/ops/triton/ssd_bmm.py