black

FANGAreNotGnu · Oct 23, 2023 · 18c0088 · 18c0088
1 parent ddcec0f
commit 18c0088
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multimodal/src/autogluon/multimodal/predictor.py b/multimodal/src/autogluon/multimodal/predictor.py
@@ -1510,7 +1510,7 @@ def _fit(
         with apply_log_filter(log_filter):
             trainer = pl.Trainer(
                 accelerator="gpu" if num_gpus > 0 else "auto",
-                devices=get_available_devices(num_gpus,config.env.auto_select_gpus),
+                devices=get_available_devices(num_gpus, config.env.auto_select_gpus),
                 num_nodes=config.env.num_nodes,
                 precision=precision,
                 strategy=strategy if strategy else "auto",