IBM-Granite-v3.1-8B-Instruct: Optimized for Mobile Deployment

State-of-the-art large language model useful on a variety of code understanding and generation tasks

Granite-3.1-8B-Instruct is a 8B parameter long-context instruct model finetuned from Granite-3.1-8B-Base using a combination of open source instruction datasets with permissive license and internally collected synthetic datasets tailored for solving long context problems. This model is developed using a diverse set of techniques with a structured chat format, including supervised finetuning, model alignment using reinforcement learning, and model merging.

This model is an implementation of IBM-Granite-v3.1-8B-Instruct found here.

More details on model performance across various devices, can be found here.

Model Details

• Model Type: Text generation
• Model Stats:
  • Input sequence length for Prompt Processor: 128
  • Context length: 4096
  • Number of parameters: 8B
  • Precision: w4a16 + w8a16 (few layers)
  • Num of key-value heads: 8
  • Information about the model parts: Prompt Processor and Token Generator are split into 5 parts each. Each corresponding Prompt Processor and Token Generator part share weights.
  • Prompt processor model size: 4.8 GB
  • Prompt processor input (part1): 128 tokens
  • Prompt processor output (part1): Embeddings output
  • Prompt processor input (other parts): 128 tokens + KVCache initialized with pad token
  • Prompt processor output (other parts): 128 output tokens + KVCache for token generator
  • Token generator model size: 4.8 GB
  • Token generator input (part1): 1 token
  • Token generator output (part1): Embeddings output
  • Token generator input (other parts): 1 input token + past KVCache
  • Token generator output (other parts): 1 output token + KVCache for next iteration
  • Use: Initiate conversation with prompt-processor and then token generator for subsequent iterations.
  • Supported natural languages: English
  • Supported programming languages: The Granite code foundation models support 116 programming languages including Python, Javascript, Java, C++, Go, and Rust.
  • Minimum QNN SDK version required: 2.3
  • TTFT: Time To First Token is the time it takes to generate the first response token. This is expressed as a range because it varies based on the length of the prompt. The lower bound is for a short prompt (up to 128 tokens, i.e., one iteration of the prompt processor) and the upper bound is for a prompt using the full context length (2048 tokens).
  • Response Rate: Rate of response generation after the first response token.

Model	Device	Chipset	Target Runtime	Response Rate (tokens per second)	Time To First Token (range, seconds)
IBM-Granite-v3.1-8B-Instruct	Snapdragon 8 Elite QRD	Snapdragon® 8 Elite	QNN	11.01293	0.19679249999999998 - 6.297359999999999
IBM-Granite-v3.1-8B-Instruct	Snapdragon X Elite CRD	Snapdragon® X Elite	QNN	8.01724	0.2953902 - 9.4524864

Deploying IBM Granite 3.1 on-device

Please follow the LLM on-device deployment tutorial.

License

• The license for the original implementation of IBM-Granite-v3.1-8B-Instruct can be found here.
• The license for the compiled assets for on-device deployment can be found here

References

• Granite Code Models: A Family of Open Foundation Models for Code Intelligence
• Source Model Implementation

Community

• Join our AI Hub Slack community to collaborate, post questions and learn more about on-device AI.
• For questions or feedback please reach out to us.

Usage and Limitations

Model may not be used for or in connection with any of the following applications:

• Accessing essential private and public services and benefits;
• Administration of justice and democratic processes;
• Assessing or recognizing the emotional state of a person;
• Biometric and biometrics-based systems, including categorization of persons based on sensitive characteristics;
• Education and vocational training;
• Employment and workers management;
• Exploitation of the vulnerabilities of persons resulting in harmful behavior;
• General purpose social scoring;
• Law enforcement;
• Management and operation of critical infrastructure;
• Migration, asylum and border control management;
• Predictive policing;
• Real-time remote biometric identification in public spaces;
• Recommender systems of social media platforms;
• Scraping of facial images (from the internet or otherwise); and/or
• Subliminal manipulation