HamSession commited on
Commit
4201429
·
verified ·
1 Parent(s): c06617d

Add modeling_stock_mamba.py

Browse files
Files changed (1) hide show
  1. 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 retired 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
  """
 
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
  """