Problem with loading to a cpu a model that was trained in CUDA

[ori-katz100]

I trained a model at AWS, saved it, but when I try to load it on my personal computer I get the following error:

~\AppData\Roaming\Python\Python37\site-packages\sdv\tabular\base.py in load(cls, path)
    559         with open(path, 'rb') as f:
--> 560             return pickle.load(f)

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

I can load models trained on my computer, and on AWS I can also load this model.

[acivit]

Hello!

Were you able to solve this bug?

Thanks!

[jinlow]

Hello curious if there was any workaround for this problem? I am also experiencing the same issue.