what version of pytorch-lightning should be used?

biomap-research / scFoundation

Apache License 2.0

224 stars 34 forks source link

Open vjcitn opened 1 month ago

vjcitn commented 1 month ago

attempts to read the models.ckpt fail with

KeyError: 'pytorch-lightning_version' ... using 2.3.3 of lightning and pytorch-lightning

WhirlFirst commented 1 month ago

Hi,

We use pytorch in our model, can you try to use pytorch for loading?