Downloads · 30 days
31
20% of all-time downloads
tokimoa/plamo-2.1-2b-vl-mlx-4bit
plamo-2.1-2b-vl-mlx-4bit is a image-text-to-text model from tokimoa. Use it for the image-text-to-text task on the model card, and read the license before you ship it in a product. It is set up for mlx. The card lists the license as other.
Preferred Networksの PLaMo 2.1-2B-VL(Mamba×Attentionハイブリッドplamo2 LM+SigLIP so400mの日英VLM)をMLX 4bit(group size 64、実効6.16bpw)に量子化した最軽量版(2.1GB・約120 tok/s)です。
Downloads · 30 days
31
20% of all-time downloads
All-time downloads
154
Public
Parameters
2.9B
2.2 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors2.2 GB · 100%
How the weights are stored.
U322.5B · 86%
From the Hugging Face model README
Preferred Networksの PLaMo 2.1-2B-VL(Mamba×Attentionハイブリッドplamo2 LM+SigLIP so400mの日英VLM)をMLX 4bit(group size 64、実効6.16bpw)に量子化した最軽量版(2.1GB・約120 tok/s)です。
当方で移植したMLX実装は**mlx-vlm本体にマージ済み(0.6.8で標準対応)**です(plamo2vl/ディレクトリ)。導入手順は下記のとおりです。
品質面の注記: 写真説明の検証プローブではbf16版と出力完全一致しますが、モデルが不確信な入力(高密度文書の読み取り等)では8bit版より出力のブレが大きいことを実測で確認しています。品質重視なら 8bit版 を推奨します。
このrepoは上流のtrust_remote_codeをtransformers 5対応に再構成済みです(上流はtransformers<5専用)。mlx-vlm 0.6.7でそのまま動きます。
mlx-vlm 0.6.8以降なら追加手順は不要です(この移植実装は#1739で本体にマージされ、0.6.8に収録されました。0.6.8での動作を実測確認済み):
pip install mlx-vlm numba torch torchvision
<details><summary>0.6.7以前を使う場合(旧手順)</summary>
repo同梱のplamo2vl/をmlx_vlm/models/へコピーしてください(詳細は旧版カード参照)。
チャットテンプレートではなく、processorが日本語Alpaca型(### 指示:/### 応答:)のプロンプトを内部で組み立てます。processor(images=画像, text=質問)だけで正しい形式になります。コード例は8bit版のカードと同じです(repo名を読み替えてください)。
同梱のMLX実装は、bf16変換版でtransformers公式実装(bf16・MPS)との突き合わせ検証を実施済みです(同一入力のprefillロジットでtop-1トークン6/6一致・cos類似度0.998以上、写真説明プローブでgreedy完全一致)。
この4bit版の検証:
用途上の注意: 2Bクラスのため高密度文書の数値読み取りは苦手です(transformers公式実装でも同様=変換起因ではありません)。写真・図の説明など軽量VL用途向きです。
pfnet/plamo-2.1-2b-vl(公式BF16、commit 3d32366)plamo2実装をベースに移植したplamo2vlパッケージ(このrepoに同梱。RoPE base=1e6のplamo-2.1系対応を含む)mlx-vlm 0.6.7 convert -q --q-bits 4(group size 64 affine、実効6.16bpw)PLaMoコミュニティライセンス(全文を同梱のLICENSE.mdに収録。本モデルの利用はその条件に従います):