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="hkqiu/PolyOpus")
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("hkqiu/PolyOpus")
model = AutoModelForCausalLM.from_pretrained("hkqiu/PolyOpus", 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

PolyOpus

PolyOpus is a polymer-specialized language model built by supervised fine-tuning of DeepSeek-LLM-7B-Chat. It serves as the specialist model in the PolySea framework.

Supported tasks

  • Predict glass-transition temperature, crystalline band gap and atomization energy.
  • Classify polyimide heat resistance using Tg-derived categories.
  • Generate polymer structures for a specified band-gap target.
  • Answer polymer-science questions in English or Chinese.

For property prediction, provide the requested property and a polymer repeat-unit SMILES. The * symbols indicate polymerization attachment points.

Model details

  • Approximately 7B parameters; LlamaForCausalLM architecture.
  • A complete merged FP16 checkpoint in three Safetensors shards. No separate LoRA adapter is required for inference.
  • A 4,096-token context window, with the tokenizer and chat template included.

Quick start

Install torch, transformers and accelerate, then run:

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "hkqiu/PolyOpus"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id, torch_dtype=torch.float16, device_map="auto"
)

messages = [{
    "role": "user",
    "content": "Predict the Tg of the following SMILES: *CC*"
}]
prompt = tokenizer.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True
)
inputs = tokenizer(
    prompt, add_special_tokens=False, return_tensors="pt"
).to(model.device)

with torch.inference_mode():
    outputs = model.generate(
        **inputs, max_new_tokens=128, do_sample=True,
        temperature=0.7, top_p=0.95,
        pad_token_id=tokenizer.eos_token_id,
    )
answer = outputs[0, inputs["input_ids"].shape[1]:]
print(tokenizer.decode(answer, skip_special_tokens=True))

Predictions and generated structures require independent validation before use in experiments or materials selection.

Links

Downloads last month
1,000
Safetensors
Model size
7B params
Tensor type
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for hkqiu/PolyOpus

Finetuned
(668)
this model