From 699314986bb2cef3a42df94ad865ab8717c28d45 Mon Sep 17 00:00:00 2001 From: Charlie Marshak Date: Tue, 4 Mar 2025 15:47:41 -0800 Subject: [PATCH] move comment around --- src/dist_s1/processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/dist_s1/processing.py b/src/dist_s1/processing.py index a46ed9c..6f1f2da 100644 --- a/src/dist_s1/processing.py +++ b/src/dist_s1/processing.py @@ -55,9 +55,9 @@ def compute_normal_params_per_burst_and_serialize( memory_strategy: str = 'high', device: str = 'best', ) -> Path: - # For distmetrics, None is how we choose the "best" available device if device not in ('cpu', 'cuda', 'mps', 'best'): raise ValueError(f'Invalid device: {device}') + # For distmetrics, None is how we choose the "best" available device if device == 'best': device = None model = load_transformer_model(device=device)