Add modeling_stock_mamba.py
Browse files- modeling_stock_mamba.py +1 -1
modeling_stock_mamba.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
"""Self-contained model definition shipped inside the Hugging Face repo.
|
| 2 |
|
| 3 |
Depends only on torch, mamba_ssm (Mamba-3) and safetensors. Mirrors the training-time
|
| 4 |
-
MambaBlock and MambaEncoder and the
|
| 5 |
state_dict loads with no dependency on the training repo. A CUDA GPU is required: the
|
| 6 |
Mamba-3 kernels have no CPU path and mamba_ssm fails at import on a CPU-only machine.
|
| 7 |
"""
|
|
|
|
| 1 |
"""Self-contained model definition shipped inside the Hugging Face repo.
|
| 2 |
|
| 3 |
Depends only on torch, mamba_ssm (Mamba-3) and safetensors. Mirrors the training-time
|
| 4 |
+
MambaBlock and MambaEncoder and the pretraining wrapper, so the exported
|
| 5 |
state_dict loads with no dependency on the training repo. A CUDA GPU is required: the
|
| 6 |
Mamba-3 kernels have no CPU path and mamba_ssm fails at import on a CPU-only machine.
|
| 7 |
"""
|