🐛 Bug
I saved a model using litmodel and then when trying to load the model, it logs a lot of irrelevant logs related to Tensorflow.
Code sample
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")
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