File size: 556 Bytes
5151bc4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
---
license: mit
language:
- en
base_model:
- ibm-granite/granite-3.2-8b-instruct
tags:
- chemistry
---

# SAC LLM

Model to predict SAC synthesis procedures.

## Usage

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