Fix: correct shape check and torch.save argument order#1

Open

pawelpratyush wants to merge 1 commit into

Helix-Research-Lab:mainfrom

pawelpratyush:fix/pooler-shape-and-save-bugs

pawelpratyush commented Apr 28, 2026

Bugs Fixed

Undefined attribute
- self.representations was referenced before assignment
- Fixed by using self.representations_with_cls.shape

Incorrect torch.save usage

Arguments were reversed:
```
torch.save(save_path, cls_token)
```
Fixed to:
```
torch.save(cls_token, save_path)
```


          Fix: correct shape check and torch.save argument order

1b23903

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet