[BUG] Loading a model doesn't work

[Theo-Clavel]

Describe the bug I trained a RNN model and saved it. When I restart my Databricks cluster and load the model, the predict function raises an error : "'Trainer' object has no attribute '_accelerator_connector'".

To Reproduce model.save("model.pt") model = RNNModel.load("model.pt") model.predict(series=ts, n=30)