--- license: mit base_model: deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B pipeline_tag: text-generation tags: - chat --- # litert-community/DeepSeek-R1-Distill-Qwen-1.5B This model provides a few variants of [deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B) that are ready for deployment on Android using the [LiteRT (fka TFLite) stack](https://ai.google.dev/edge/litert), [MediaPipe LLM Inference API](https://ai.google.dev/edge/mediapipe/solutions/genai/llm_inference) and [LiteRt-LM](https://github.com/google-ai-edge/LiteRT-LM). ## Use the models ### Colab *Disclaimer: The target deployment surface for the LiteRT models is Android/iOS/Web and the stack has been optimized for performance on these targets. Trying out the system in Colab is an easier way to familiarize yourself with the LiteRT stack, with the caveat that the performance (memory and latency) on Colab could be much worse than on a local device.* [](https://colab.research.google.com/#fileId=https://huggingface.co/litert-community/DeepSeek-R1-Distill-Qwen-1.5B/blob/main/notebook.ipynb) ### Android #### Edge Gallery App * Download or build the [app](https://github.com/google-ai-edge/gallery?tab=readme-ov-file#-get-started-in-minutes) from GitHub. * Install the [app](https://play.google.com/store/apps/details?id=com.google.ai.edge.gallery&pli=1) from Google Play * Follow the instructions in the app. #### LLM Inference API * Download and install [the apk](https://github.com/google-ai-edge/mediapipe-samples/releases/latest/download/llm_inference-debug.apk). * Follow the instructions in the app. To build the demo app from source, please follow the [instructions](https://github.com/google-ai-edge/mediapipe-samples/blob/main/examples/llm_inference/android/README.md) from the GitHub repository. ## Performance ### Android Note that all benchmark stats are from a Samsung S24 Ultra with 1280 KV cache size with multiple prefill signatures enabled.
Backend | Quantization | Context Length | Prefill (tokens/sec) | Decode (tokens/sec) | Time-to-first-token (sec) | Model size (MB) | Peak RSS Memory (MB) | GPU Memory (MB) |
---|---|---|---|---|---|---|---|---|
CPU |
dynamic_int8 |
4096 |
166.50 tk/s |
26.35 tk/s |
6.41 s |
1831.43 MB |
2221 MB |
N/A |
GPU |
dynamic_int8 |
4096 |
927.54 tk/s |
26.98 tk/s |
5.46 s |
1831.43 MB |
2096 MB |
1659 MB |