pippy.SaveModule not exist？

[OpenSaozi]

In examples/checkpoint/toy_model.py

`from pippy.compile import compile_stage

from pippy.SaveModule import save_checkpoint`

which is not exist?

[yi-w11]

I have the same problem. Have you solved it?

[OpenSaozi]

I do found some likely file/lib in the different path. Through some hacky work, it should work...

but what I want to say is, this function is not programmingly structured well, also code is hard to read understand. Many bugs like this one. The improvement at performence is not good on my sever. My own implemention of interpipe/gpipe with single thread can be fast than pippy...