floneum/kalosm
Instant, controllable, local pre-trained AI models in Rust
它解決的問題
Kalosm 是一個基於 Rust 的生態系統,旨在簡化整合本地或遠端 AI 模型的應用程式開發。它提供了一個統一的介面,用於與各種預訓練模型互動,這些模型涵蓋不同模式(文字、音訊和影像),並包含用於結構化資料生成和 LLM 上下文管理的工具。
如何運作
該專案由兩個主要組件組成:
- Kalosm:預訓練模型的高階介面。支援 Llama、Mistral、Phi、Whisper、Segment Anything 和 Bert 等模型。包含從各種檔案格式(PDF、HTML、DOCX 等)提取上下文的實用工具、文字分塊功能,以及與向量資料庫整合以進行語意搜尋。
- Fusor:基於 WGPU 的量化機器學習推理執行時。使用 GGUF 檔案格式,並透過核心融合編譯器將操作鏈優化為單一核心,使模型能原生運行於各種加速器(Nvidia、AMD、Metal)或直接在瀏覽器中透過 WebGPU 執行。
適用對象
希望使用一致 API、支援本地量化模型,並在多樣硬體加速器上高效執行模型的 Rust 開發者。
主要特色
- 多模態支援:整合支援文字(LLMs)、音訊(Whisper)和影像(Segment Anything)模型。
- 結構化生成:使用自訂解析引擎和取樣器,透過基於語法的約束,將模型輸出限制為特定的 Rust 類型、JSON、HTML 或 XML。
- Firestore 整合:提供用於上下文提取、網頁爬取和資料擷取的實用工具,以供 LLM 使用。
- 硬體無關:透過 Fusor,可針對多種 GPU 架構和瀏覽器使用 WGPU。
- 量化推理:原生支援量化模型,以降低記憶體和運算需求。
相關
- 專案
- 專案
- 專案
- 專案
- 專案