Use dynamic CUDA check instead of hardcoded device in emb_extractor.py
#593
by dlgenomics - opened
Fixes CPU-only inference, which currently fails with "AssertionError: Torch not compiled with CUDA enabled" because device="cuda" is hardcoded.
Replaces the hardcoded "cuda" with a dynamic check ("cuda" if torch.cuda.is_available() else "cpu"), matching the pattern already used elsewhere in the codebase (e.g. perturber_utils.py line 194).
See discussion #592: https://huggingface.co/ctheodoris/Geneformer/discussions/592