drugilsberg commited on
Commit
5151bc4
·
verified ·
1 Parent(s): a2b784e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ base_model:
6
+ - ibm-granite/granite-3.2-8b-instruct
7
+ tags:
8
+ - chemistry
9
+ ---
10
+
11
+ # SAC LLM
12
+
13
+ Model to predict SAC synthesis procedures.
14
+
15
+ ## Usage
16
+
17
+ ```python
18
+ from transformers import AutoTokenizer, AutoModelForCausalLM
19
+
20
+ model_id = "rxn4chemistry/sac-llm-ft-multitask"
21
+ tok = AutoTokenizer.from_pretrained(model_id)
22
+ model = AutoModelForCausalLM.from_pretrained(model_id)
23
+
24
+ prompt = "Hello!"
25
+ inputs = tok(prompt, return_tensors="pt")
26
+ out = model.generate(**inputs, max_new_tokens=64)
27
+ print(tok.decode(out[0], skip_special_tokens=True))
28
+ ``