This repository has been archived by the owner on Oct 9, 2024. It is now read-only.

Remove to_device #85

Open

canergen opened this issue Feb 8, 2024 · 2 comments

Member

canergen commented Feb 8, 2024

@martinkim0 Is this handled now after adding scvi-tools>1.0.0 support?
https://github.com/YosefLab/scvi-v2/blob/7f4984a8371c0ee43d60487bb280322dcd9a50f5/src/scvi_v2/_model.py#L134

Member

martinkim0 commented Feb 8, 2024

I think so now that we have #82 merged, although that method isn't doing anything (?).

Member Author

canergen commented Feb 8, 2024

Yes, it was a placeholder. I remember to_device wasn't set up in Jax before we added the arguments to scvi-tools.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.