Skip to content

Commit

Permalink
move comment around
Browse files Browse the repository at this point in the history
  • Loading branch information
cmarshak committed Mar 4, 2025
1 parent 8c2113d commit 6993149
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/dist_s1/processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 6993149

Please sign in to comment.