drugilsberg's picture
Create README.md
5151bc4 verified
metadata
license: mit
language:
  - en
base_model:
  - ibm-granite/granite-3.2-8b-instruct
tags:
  - chemistry

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))
``