# Just test and read the activation pt file import torch ids = torch.load(f'./ids/qwen2.5-0.5/id.eu.train.qwen2.5-0.5') print(ids.shape)