diff --git a/swarms/utils/load_model_torch.py b/swarms/utils/load_model_torch.py index 53649e93..113ff966 100644 --- a/swarms/utils/load_model_torch.py +++ b/swarms/utils/load_model_torch.py @@ -1,3 +1,6 @@ +""" Load a PyTorch model from a given path and move it to the specified device. """ +# flake8: noqa E501 + import torch from torch import nn