Update README.md
Browse files
README.md
CHANGED
|
@@ -14,8 +14,8 @@ information and tips on how to best run the model.
|
|
| 14 |
|
| 15 |
## Usage
|
| 16 |
The default adapter uses cosine distance training, while an alternative variant
|
| 17 |
-
trained with MSE can be loaded by specifying
|
| 18 |
-
`load_adapter` function.
|
| 19 |
|
| 20 |
```python
|
| 21 |
from detikzify.model import load, load_adapter
|
|
@@ -30,7 +30,7 @@ pipeline = DetikzifyPipeline(
|
|
| 30 |
torch_dtype="bfloat16",
|
| 31 |
),
|
| 32 |
adapter_name_or_path="nllg/tikzero-adapter",
|
| 33 |
-
|
| 34 |
)
|
| 35 |
)
|
| 36 |
|
|
|
|
| 14 |
|
| 15 |
## Usage
|
| 16 |
The default adapter uses cosine distance training, while an alternative variant
|
| 17 |
+
trained with MSE can be loaded by specifying
|
| 18 |
+
`adapter_kwargs=dict(revision="mse")` in the `load_adapter` function.
|
| 19 |
|
| 20 |
```python
|
| 21 |
from detikzify.model import load, load_adapter
|
|
|
|
| 30 |
torch_dtype="bfloat16",
|
| 31 |
),
|
| 32 |
adapter_name_or_path="nllg/tikzero-adapter",
|
| 33 |
+
adapter_kwargs=dict(revision="mse") # load variant trained with MSE
|
| 34 |
)
|
| 35 |
)
|
| 36 |
|