[Speedster] `optimize_model` function transfers the original model from the GPU to the CPU

[mfumanelli]

Problem

The optimize_model function transfers the original model from the GPU to the CPU:

Desired Behavior

Maintaining the model on the initial device.

[LuigiCerone]

I can try to work on this @mfumanelli! 🌈

[mfumanelli]

I can try to work on this @mfumanelli! 🌈

[youssefadr]

Hello @LuigiCerone, I see there was no PR related to this issue in the previous months, are you still planning to work on this? If not can I tackle this issue @mfumanelli?

[LuigiCerone]

Hello @youssefadr, go for it! Thanks :)

[youssefadr]

Great, I'll make a PR soon!

[youssefadr]

I am working on it and debugging to evaluate where the bug comes from, if someone has already worked on it and has more insights, I'll be happy to listen!