Instructions to use NingLab/GeLLMO-P4-Mistral with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use NingLab/GeLLMO-P4-Mistral with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="NingLab/GeLLMO-P4-Mistral")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("NingLab/GeLLMO-P4-Mistral", device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use NingLab/GeLLMO-P4-Mistral with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "NingLab/GeLLMO-P4-Mistral" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "NingLab/GeLLMO-P4-Mistral", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/NingLab/GeLLMO-P4-Mistral
- SGLang
How to use NingLab/GeLLMO-P4-Mistral 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 "NingLab/GeLLMO-P4-Mistral" \ --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": "NingLab/GeLLMO-P4-Mistral", "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 "NingLab/GeLLMO-P4-Mistral" \ --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": "NingLab/GeLLMO-P4-Mistral", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use NingLab/GeLLMO-P4-Mistral with Docker Model Runner:
docker model run hf.co/NingLab/GeLLMO-P4-Mistral
| license: cc-by-nc-4.0 | |
| datasets: | |
| - NingLab/MuMOInstruct | |
| language: | |
| - en | |
| base_model: | |
| - mistralai/Mistral-7B-Instruct-v0.3 | |
| pipeline_tag: text-generation | |
| tags: | |
| - chemistry | |
| - molecule optimization | |
| library_name: transformers | |
| ### Model Sources | |
| <!-- Provide the basic links for the model. --> | |
| - **Repository:** https://github.com/ninglab/GeLLMO | |
| - **Paper:** https://arxiv.org/abs/2502.13398 | |
| ## Usage | |
| For instructions to run the model, please refer to our repository. | |
| ## Bias, Risks, and Limitations | |
| While our models are designed for research and drug discovery applications, | |
| they come with ethical and safety considerations: | |
| 1. **Potential for Misuse:** Although the model is not explicitly designed to generate toxic, | |
| controlled, or harmful compounds, adversarial prompts or unintended biases in the pretrained model | |
| may lead to the generation of undesirable molecules. | |
| 2. **Unintended Harmful Outputs:** The model does not inherently filter out molecules with high toxicity, | |
| abuse potential, or environmental hazards. Users must implement additional safeguards to prevent misuse. | |
| 3. **Absence of Built-in Safety Mechanisms:** The model does not incorporate explicit regulatory or | |
| safety filters (e.g., toxicity or compliance checks). | |
| It is the responsibility of users to validate generated molecules for safety and ethical considerations. | |
| We urge users to adopt best practices, including toxicity prediction pipelines, | |
| ethical oversight, and responsible AI usage policies, to prevent harmful applications of this model. | |
| ## Citation | |
| If you use the trained model checkpoints, datasets or other resources, please use the following citation: | |
| ``` | |
| @article{dey2025gellmo, | |
| title={GeLLMO: Generalizing Large Language Models for Multi-property Molecule Optimization}, | |
| author={Vishal Dey and Xiao Hu and Xia Ning}, | |
| year={2025}, | |
| journal={arXiv preprint arXiv:2502.13398}, | |
| url={https://arxiv.org/abs/2502.13398}, | |
| } | |
| ``` |