Views
No views yet
LipoAgent: Coordinating Fine-Tuned LLM Agents for Safer Lipid Design
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5base_model_id = "Qwen/Qwen3-8B"
6adapter_id = "Nemo0412/qwen3-8b-lipo"
7
8tokenizer = AutoTokenizer.from_pretrained(base_model_id)
9base_model = AutoModelForCausalLM.from_pretrained(
10 base_model_id, torch_dtype=torch.bfloat16, device_map="auto"
11)
12model = PeftModel.from_pretrained(base_model, adapter_id)