Neel Rajani

icon indicating a website link https://neelectric.github.io/

icon company University of Edinburgh icon location Edinburgh

Results 1 issues of Neel Rajani

Make lenses device agnostic using map_location in torch.load()

3 comment

Some .pt files on the huggingface space (for eg [this one](https://huggingface.co/spaces/AlignmentResearch/tuned-lens/tree/main/lens/EleutherAI/pythia-410m-deduped), looks like all pythia lenses might be affected? gpt2-large seems to work) contain "cuda:0". See the following screenshot (forgive...