Code

Add an MLX implementation for mimi#230

Code

Merged

LaurentMazare merged 23 commits intomainkyutai-labs/moshi:mainfrom mlx-mimikyutai-labs/moshi:mlx-mimiCopy head branch name to clipboardFeb 22, 2025

Commits

Commits on Feb 20, 2025

• Start adding mimi to the mlx codebase.

  LaurentMazare

  committed
  da3d127

  Copy full SHA for da3d127

• Finish the conv module.

  LaurentMazare

  committed
  a0971b6

  Copy full SHA for a0971b6

• Start adding the seanet module.

  LaurentMazare

  committed
  09fdef1

  Copy full SHA for 09fdef1

• More seanet.

  LaurentMazare

  committed
  fc7c728

  Copy full SHA for fc7c728

• Add to seanet.

  LaurentMazare

  committed
  1395d2f

  Copy full SHA for 1395d2f

• Start adding some quantization.

  LaurentMazare

  committed
  998adca

  Copy full SHA for 998adca

• More quantization support.

  LaurentMazare

  committed
  d0659a0

  Copy full SHA for d0659a0

Commits on Feb 21, 2025

• Formatting.

  LaurentMazare

  committed
  a8ef530

  Copy full SHA for a8ef530

• Make pyright happy.

  LaurentMazare

  committed
  565ba71

  Copy full SHA for 565ba71

• Merge remote-tracking branch 'origin/main' into mlx-mimi

  LaurentMazare

  committed
  afff244

  Copy full SHA for afff244

• Add the mimi config.

  LaurentMazare

  committed
  06ec202

  Copy full SHA for 06ec202

• More mimi implementation.

  LaurentMazare

  committed
  9a1b25f

  Copy full SHA for 9a1b25f

• Add a test script.

  LaurentMazare

  committed
  9e9a17e

  Copy full SHA for 9e9a17e

• Shape bugfixes.

  LaurentMazare

  committed
  2664857

  Copy full SHA for 2664857

• Bugfixes...

  LaurentMazare

  committed
  bfcb20d

  Copy full SHA for bfcb20d

• Again more mimi.

  LaurentMazare

  committed
  61d36df

  Copy full SHA for 61d36df

• Get some roundtrip to work.

  LaurentMazare

  committed
  9db07c2

  Copy full SHA for 9db07c2

Commits on Feb 22, 2025

• Improve the weight loading.

  LaurentMazare

  committed
  29d7fe8

  Copy full SHA for 29d7fe8

• Hacky import.

  LaurentMazare

  committed
  84fff57

  Copy full SHA for 84fff57

• Get the weight loading to work.

  LaurentMazare

  committed
  d937390

  Copy full SHA for d937390

• Write the generated file.

  LaurentMazare

  committed
  746cc3d

  Copy full SHA for 746cc3d

• Rename the hacky weight loader.

  LaurentMazare

  committed
  e9cfc76

  Copy full SHA for e9cfc76

• Fix the pre-commit issue.

  LaurentMazare

  committed
  5d938b6

  Copy full SHA for 5d938b6