rxn4chemistry/sac-llm-ft-multitask
06
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))
``