Update README.md
Browse files
README.md
CHANGED
|
@@ -31,6 +31,7 @@ pipeline_tag: text-generation
|
|
| 31 |
# Inference Code:
|
| 32 |
|
| 33 |
- Feel free to make the steps and verifications collapsable and the initial reasoning too, you can show only the final answer to get an o1 feel(i don't know)
|
|
|
|
| 34 |
|
| 35 |
```python
|
| 36 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
| 31 |
# Inference Code:
|
| 32 |
|
| 33 |
- Feel free to make the steps and verifications collapsable and the initial reasoning too, you can show only the final answer to get an o1 feel(i don't know)
|
| 34 |
+
- **Note:** A feature we have here is the ability to control how many steps and verifications you want.
|
| 35 |
|
| 36 |
```python
|
| 37 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|