Upload FAST tokenizer trained on lerobot/libero_video
Browse files
processing_action_tokenizer.py
CHANGED
|
@@ -16,7 +16,7 @@ class WaveletActionProcessor(ProcessorMixin):
|
|
| 16 |
def __init__(
|
| 17 |
self,
|
| 18 |
bpe_tokenizer: PreTrainedTokenizerFast,
|
| 19 |
-
wavelet: str = "
|
| 20 |
level: int = 2,
|
| 21 |
scale: float = 10.0,
|
| 22 |
min_token: int = 0,
|
|
|
|
| 16 |
def __init__(
|
| 17 |
self,
|
| 18 |
bpe_tokenizer: PreTrainedTokenizerFast,
|
| 19 |
+
wavelet: str = "db4",
|
| 20 |
level: int = 2,
|
| 21 |
scale: float = 10.0,
|
| 22 |
min_token: int = 0,
|