├── .gitignore ├── FINETUNING.md ├── FINETUNING_CLI.md ├── LICENSE ├── README.md ├── csm_mlx ├── __init__.py ├── __main__.py ├── attention.py ├── cli │ ├── __init__.py │ ├── application.py │ ├── config.py │ ├── finetune │ │ ├── __init__.py │ │ ├── dataset.py │ │ ├── full_finetune.py │ │ ├── lora_finetune.py │ │ └── utils.py │ └── generate.py ├── config.py ├── finetune │ ├── __init__.py │ ├── dataset.py │ ├── trainer.py │ └── utils.py ├── generation.py ├── models.py ├── segment.py ├── tokenizers.py └── utils.py └── pyproject.toml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/.gitignore -------------------------------------------------------------------------------- /FINETUNING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/FINETUNING.md -------------------------------------------------------------------------------- /FINETUNING_CLI.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/FINETUNING_CLI.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/README.md -------------------------------------------------------------------------------- /csm_mlx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/__init__.py -------------------------------------------------------------------------------- /csm_mlx/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/__main__.py -------------------------------------------------------------------------------- /csm_mlx/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/attention.py -------------------------------------------------------------------------------- /csm_mlx/cli/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /csm_mlx/cli/application.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/application.py -------------------------------------------------------------------------------- /csm_mlx/cli/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/config.py -------------------------------------------------------------------------------- /csm_mlx/cli/finetune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/finetune/__init__.py -------------------------------------------------------------------------------- /csm_mlx/cli/finetune/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/finetune/dataset.py -------------------------------------------------------------------------------- /csm_mlx/cli/finetune/full_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/finetune/full_finetune.py -------------------------------------------------------------------------------- /csm_mlx/cli/finetune/lora_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/finetune/lora_finetune.py -------------------------------------------------------------------------------- /csm_mlx/cli/finetune/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/finetune/utils.py -------------------------------------------------------------------------------- /csm_mlx/cli/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/cli/generate.py -------------------------------------------------------------------------------- /csm_mlx/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/config.py -------------------------------------------------------------------------------- /csm_mlx/finetune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/finetune/__init__.py -------------------------------------------------------------------------------- /csm_mlx/finetune/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/finetune/dataset.py -------------------------------------------------------------------------------- /csm_mlx/finetune/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/finetune/trainer.py -------------------------------------------------------------------------------- /csm_mlx/finetune/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/finetune/utils.py -------------------------------------------------------------------------------- /csm_mlx/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/generation.py -------------------------------------------------------------------------------- /csm_mlx/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/models.py -------------------------------------------------------------------------------- /csm_mlx/segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/segment.py -------------------------------------------------------------------------------- /csm_mlx/tokenizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/tokenizers.py -------------------------------------------------------------------------------- /csm_mlx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/csm_mlx/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/senstella/csm-mlx/HEAD/pyproject.toml --------------------------------------------------------------------------------