docs: add library_name: invariantone and clarify from_pretrained alias in quickstart
Browse files- MODEL_CARD.md +5 -0
MODEL_CARD.md
CHANGED
|
@@ -5,6 +5,7 @@ license: apache-2.0
|
|
| 5 |
base_model:
|
| 6 |
- Qwen/Qwen3.5-4B-Base
|
| 7 |
pipeline_tag: text-classification
|
|
|
|
| 8 |
tags:
|
| 9 |
- decision-model
|
| 10 |
- permutation-equivariance
|
|
@@ -34,6 +35,7 @@ pip install invariantone==1.0.2
|
|
| 34 |
```python
|
| 35 |
from invariantone import InvariantOne
|
| 36 |
|
|
|
|
| 37 |
model = InvariantOne.from_pretrained("invariantone-v1")
|
| 38 |
|
| 39 |
result = model.decide(
|
|
@@ -51,6 +53,9 @@ print(result.choice)
|
|
| 51 |
print(result.probabilities)
|
| 52 |
```
|
| 53 |
|
|
|
|
|
|
|
|
|
|
| 54 |
---
|
| 55 |
|
| 56 |
## 1. Model Overview
|
|
|
|
| 5 |
base_model:
|
| 6 |
- Qwen/Qwen3.5-4B-Base
|
| 7 |
pipeline_tag: text-classification
|
| 8 |
+
library_name: invariantone
|
| 9 |
tags:
|
| 10 |
- decision-model
|
| 11 |
- permutation-equivariance
|
|
|
|
| 35 |
```python
|
| 36 |
from invariantone import InvariantOne
|
| 37 |
|
| 38 |
+
# Load packaged v1 weights (canonical package alias for TanmaySah/InvariantOne-v1)
|
| 39 |
model = InvariantOne.from_pretrained("invariantone-v1")
|
| 40 |
|
| 41 |
result = model.decide(
|
|
|
|
| 53 |
print(result.probabilities)
|
| 54 |
```
|
| 55 |
|
| 56 |
+
> **Note on Model Identifier:**
|
| 57 |
+
> `"invariantone-v1"` is the official package alias that resolves the verified weights bundled directly inside `invariantone`. If you clone this repository locally, you can also load directly from the local directory via `InvariantOne.from_pretrained("./")`.
|
| 58 |
+
|
| 59 |
---
|
| 60 |
|
| 61 |
## 1. Model Overview
|