NotImplementedError: Loading model from .pth is not supported yet.

[deependujha]

🐛 Bug

LitModels saved pytorch model in .pth format, but doesn't support loading it back.

import litmodels as lm
import torch

# save a model
model = torch.nn.Module()
# lm.upload_model(model=model, name="deep-litmodel-1")

# load a model
model = lm.load_model(name="deep-litmodel-1")

• output:

Downloading default: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████| 924/924 [00:00<00:00, 2.09kB/s]
Traceback (most recent call last):                                                                                                     | 0.00/924 [00:00<?, ?B/s]
  File "/teamspace/studios/this_studio/main.py", line 9, in <module>
    model = lm.load_model(name="deep-litmodel-1")
  File "/teamspace/studios/this_studio/LitModels/src/litmodels/io/gateway.py", line 123, in load_model
    raise NotImplementedError(f"Loading model from {model_path.suffix} is not supported yet.")
NotImplementedError: Loading model from .pth is not supported yet.

Code sample

Expected behavior

Environment

• PyTorch Version (e.g., 1.0):
• OS (e.g., Linux):
• How you installed PyTorch (conda, pip, source):
• Build command you used (if compiling from source):
• Python version:
• CUDA/cuDNN version:
• GPU models and configuration:
• Any other relevant information:

Additional context

[deependujha]

this makes sense, but then readme toy example is incorrect.