Skip to content

Commit 6a4cb18

Browse files
committed
added demo video
Merge branch 'main' of github.com:jasonkena/magicpickle add gotchas to README.md
1 parent f570139 commit 6a4cb18

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
# MagicPickle
2+
[magicpickle_demo.webm](https://github.com/user-attachments/assets/5db0f63c-f44b-485d-b67e-97266f691d99)
3+
24
`magicpickle` allows you to transfer pickled representations of objects between local and remote instances of scripts, providing a near-seamless way to write code which both accesses data stored remotely and visualizes it locally. This avoids the need to:
35
- store, load, and sync intermediate data representations between local and remote machines
46
- use X11 forwarding/VNC with noticable latency
@@ -37,7 +39,7 @@ import dill as pickle
3739
```
3840

3941
To allow the loading of pickled CUDA tensors onto a CPU, prefix your script with
40-
```
42+
```python
4143
import torch
4244
# https://stackoverflow.com/a/78399538/10702372
4345
torch.serialization.register_package(0, lambda x: x.device.type, lambda x, _: x.cpu())

0 commit comments

Comments
 (0)