Skip to content

Commit 0eb43ae

Browse files
committed
deps: move dflash-mlx from optional to required dependency
1 parent 3584025 commit 0eb43ae

2 files changed

Lines changed: 3 additions & 9 deletions

File tree

omlx/engine/__init__.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,7 @@
1717
from .batched import BatchedEngine
1818
from .embedding import EmbeddingEngine
1919

20-
# DFlash engine (optional: requires dflash-mlx)
21-
try:
22-
from .dflash import DFlashEngine
23-
except ImportError:
24-
DFlashEngine = None # type: ignore
20+
from .dflash import DFlashEngine
2521
from .reranker import RerankerEngine
2622
from .stt import STTEngine
2723
from .sts import STSEngine

pyproject.toml

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ dependencies = [
6464
# mlx-vlm from commit (23e1dff) - Gemma 4 multi-image fix + nested tool parser
6565
"mlx-vlm @ git+https://github.com/Blaizzy/mlx-vlm@23e1dffd224488141a4f022b6d21d6a730f11507",
6666
"Pillow>=9.0.0",
67+
# dflash-mlx from commit (fc7101b) — block diffusion speculative decoding
68+
"dflash-mlx @ git+https://github.com/bstnxbt/dflash-mlx@fc7101b0aa70c60f0754f50a7a10a82c2cdc1a92",
6769
]
6870

6971
[project.optional-dependencies]
@@ -77,10 +79,6 @@ mcp = [
7779
modelscope = [
7880
"modelscope>=1.10.0",
7981
]
80-
dflash = [
81-
# dflash-mlx from commit (fc7101b) — block diffusion speculative decoding
82-
"dflash-mlx @ git+https://github.com/bstnxbt/dflash-mlx@fc7101b0aa70c60f0754f50a7a10a82c2cdc1a92",
83-
]
8482
audio = [
8583
# mlx-audio from commit (5175326) with tts/stt/sts extras
8684
# tts extra: misaki, num2words, spacy, phonemizer-fork, espeakng-loader, sentencepiece

0 commit comments

Comments
 (0)