Instructions to use MBZUAI/LaMini-GPT-1.5B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use MBZUAI/LaMini-GPT-1.5B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="MBZUAI/LaMini-GPT-1.5B")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("MBZUAI/LaMini-GPT-1.5B") model = AutoModelForCausalLM.from_pretrained("MBZUAI/LaMini-GPT-1.5B") - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use MBZUAI/LaMini-GPT-1.5B with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "MBZUAI/LaMini-GPT-1.5B" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "MBZUAI/LaMini-GPT-1.5B", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/MBZUAI/LaMini-GPT-1.5B
- SGLang
How to use MBZUAI/LaMini-GPT-1.5B with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "MBZUAI/LaMini-GPT-1.5B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "MBZUAI/LaMini-GPT-1.5B", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "MBZUAI/LaMini-GPT-1.5B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "MBZUAI/LaMini-GPT-1.5B", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use MBZUAI/LaMini-GPT-1.5B with Docker Model Runner:
docker model run hf.co/MBZUAI/LaMini-GPT-1.5B
Any plan to integrate with GPTQ?
#6
by changtimwu - opened
Hi! LaMini Team:
First of all, thanks for the great work. While beyond Alpaca with only 1.5B parameters is indeed quite a breakthrough, have you consider making it even “mini” by applying the well-known GPTQ technique on it? I know that’s more about the implementation side work. It’s still useful for embedded system where an LM is expected to be intelligent instead of informative.
changtimwu changed discussion title from Any plan to integrate with GPTQ to Any plan to integrate with GPTQ?
Try AutoGPTQ, so that you don't have to write your own GPTQ implementation