How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="blazerye/DrugAssist-7B")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("blazerye/DrugAssist-7B")
model = AutoModelForCausalLM.from_pretrained("blazerye/DrugAssist-7B", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

🐹 DrugAssist

A Large Language Model for Molecule Optimization

📃 Paper • 🤗 Dataset

Please refer to our repository and paper for more details.

Downloads last month
109
GGUF
Model size
7B params
Architecture
llama
Hardware compatibility
Log In to add your hardware

We're not able to determine the quantization variants.

Inference Providers NEW
Input a message to start chatting with blazerye/DrugAssist-7B.

Model tree for blazerye/DrugAssist-7B

Quantizations
1 model