fix: hilarious typo

Signed-off-by: thiswillbeyourgithub <26625900+thiswillbeyourgithub@users.noreply.github.com>
This commit is contained in:
thiswillbeyourgithub
2025-09-11 08:33:30 +02:00
parent 3934a4219b
commit 3e86c6a1a8
+1 -1
View File
@@ -33,7 +33,7 @@ f
# num_rows: 20
# })
# it has this sgaoe
# it has this shape
ds_a = Dataset.from_parquet(str(f)).with_format("torch")
ds_a[0:2]['hidden_states'].shape # [batch, layers, tokens, hidden_states]
# torch.Size([2, 25, 1, 896])