bug: CUDA fix in decipher

[ori-kron-wis]

@justjhong I already fixed one of the tests (but you should review it)

close #3109

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 83.27%. Comparing base (6ae39a4) to head (013835c).
Report is 2 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3110      +/-   ##
==========================================
- Coverage   87.67%   83.27%   -4.40%     
==========================================
  Files         180      180              
  Lines       15187    15187              
==========================================
- Hits        13315    12647     -668     
- Misses       1872     2540     +668     

Files with missing lines	Coverage Δ
src/scvi/external/decipher/_model.py	96.11% <100.00%> (ø)

... and 11 files with indirect coverage changes

[justjhong]

Sorry about this and thanks for squashing the bug! Is the other test fixed as well or should I take a stab at it?

[canergen]

Hi, I don't think so the test reports where off here, so I restarted them. It's pretty much the same thing with device handling.

[ori-kron-wis]

@justjhong no the other test is failing, I wasnt sure what will be the way to go, please see if you can fix it.
yes it is basically on device handling (cpu and gpu calculation at the same time, but in the trajectory part, which is all cpu)