rxn4chemistry/sac-llm-ft-multitask

SAC LLM

Model to predict SAC synthesis procedures.

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM

model_id = "rxn4chemistry/sac-llm-ft-multitask"
tok = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

prompt = "Hello!"
inputs = tok(prompt, return_tensors="pt")
out = model.generate(**inputs, max_new_tokens=64)
print(tok.decode(out[0], skip_special_tokens=True))
``