potamides commited on
Commit
a6fa121
·
verified ·
1 Parent(s): 2647d59

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 `variant="mse"` in the
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
- #variant="mse" # load variant trained with MSE
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