gitextract_9ebfdj2x/ ├── .gitignore ├── LICENSE ├── README.md ├── ckpts/ │ ├── SEMamba_advanced.pth │ ├── config.yaml │ ├── pretrained_discriminator.pth │ └── vd.pth ├── data/ │ ├── make_dataset_json.py │ ├── test_clean.json │ ├── test_noisy.json │ ├── train_clean.json │ ├── train_noisy.json │ ├── valid_clean.json │ └── valid_noisy.json ├── dataloaders/ │ └── dataloader_vctk.py ├── inference.py ├── make_dataset.sh ├── mamba-1_2_0_post1/ │ ├── .github/ │ │ └── workflows/ │ │ └── publish.yaml │ ├── .gitignore │ ├── .gitmodules │ ├── AUTHORS │ ├── LICENSE │ ├── README.md │ ├── benchmarks/ │ │ └── benchmark_generation_mamba_simple.py │ ├── csrc/ │ │ └── selective_scan/ │ │ ├── reverse_scan.cuh │ │ ├── selective_scan.cpp │ │ ├── selective_scan.h │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ ├── selective_scan_bwd_bf16_real.cu │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ ├── selective_scan_bwd_fp16_real.cu │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ ├── selective_scan_bwd_fp32_real.cu │ │ ├── selective_scan_bwd_kernel.cuh │ │ ├── selective_scan_common.h │ │ ├── selective_scan_fwd_bf16.cu │ │ ├── selective_scan_fwd_fp16.cu │ │ ├── selective_scan_fwd_fp32.cu │ │ ├── selective_scan_fwd_kernel.cuh │ │ ├── static_switch.h │ │ └── uninitialized_copy.cuh │ ├── evals/ │ │ └── lm_harness_eval.py │ ├── mamba_ssm/ │ │ ├── __init__.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── config_mamba.py │ │ │ └── mixer_seq_simple.py │ │ ├── modules/ │ │ │ ├── __init__.py │ │ │ └── mamba_simple.py │ │ ├── ops/ │ │ │ ├── __init__.py │ │ │ ├── selective_scan_interface.py │ │ │ └── triton/ │ │ │ ├── __init__.py │ │ │ ├── layernorm.py │ │ │ └── selective_state_update.py │ │ └── utils/ │ │ ├── __init__.py │ │ ├── generation.py │ │ └── hf.py │ ├── setup.py │ └── tests/ │ └── ops/ │ ├── test_selective_scan.py │ └── triton/ │ └── test_selective_state_update.py ├── mamba_install/ │ ├── AUTHORS │ ├── LICENSE │ ├── README.md │ ├── benchmarks/ │ │ └── benchmark_generation_mamba_simple.py │ ├── csrc/ │ │ └── selective_scan/ │ │ ├── reverse_scan.cuh │ │ ├── selective_scan.cpp │ │ ├── selective_scan.h │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ ├── selective_scan_bwd_bf16_real.cu │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ ├── selective_scan_bwd_fp16_real.cu │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ ├── selective_scan_bwd_fp32_real.cu │ │ ├── selective_scan_bwd_kernel.cuh │ │ ├── selective_scan_common.h │ │ ├── selective_scan_fwd_bf16.cu │ │ ├── selective_scan_fwd_fp16.cu │ │ ├── selective_scan_fwd_fp32.cu │ │ ├── selective_scan_fwd_kernel.cuh │ │ ├── static_switch.h │ │ └── uninitialized_copy.cuh │ ├── evals/ │ │ └── lm_harness_eval.py │ ├── mamba_ssm/ │ │ ├── __init__.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── config_mamba.py │ │ │ └── mixer_seq_simple.py │ │ ├── modules/ │ │ │ ├── __init__.py │ │ │ └── mamba_simple.py │ │ ├── ops/ │ │ │ ├── __init__.py │ │ │ ├── selective_scan_interface.py │ │ │ └── triton/ │ │ │ ├── __init__.py │ │ │ ├── layernorm.py │ │ │ └── selective_state_update.py │ │ └── utils/ │ │ ├── __init__.py │ │ ├── generation.py │ │ └── hf.py │ ├── setup.py │ └── tests/ │ └── ops/ │ ├── test_selective_scan.py │ └── triton/ │ └── test_selective_state_update.py ├── models/ │ ├── codec_module.py │ ├── discriminator.py │ ├── generator.py │ ├── loss.py │ ├── lsigmoid.py │ ├── mamba_block.py │ ├── pcs400.py │ └── stfts.py ├── pretrained.sh ├── recipes/ │ ├── SEMamba_advanced/ │ │ ├── SEMamba_advanced.yaml │ │ └── SEMamba_advanced_pretrainedD.yaml │ └── SEMamba_advanced_PCS/ │ └── SEMamba_advanced_PCS.yaml ├── requirements.txt ├── run.sh ├── runPCS.sh ├── train.py └── utils/ └── util.py