File size: 311 Bytes
f3b0016 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from transformers import AutoProcessor
import numpy as np
processor = AutoProcessor.from_pretrained(
"/home/jade_choghari/robot/lerobot/outputs/resfast",
)
x = np.random.rand(2, 50, 14).astype("float32")
tokens = processor(x)
decoded = processor.decode(tokens)
print(tokens[0][:10])
print(decoded.shape)
|