From 1a1b79a16d2e5d7cc7e8865350de0e8afde24357 Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 24 May 2023 12:44:26 -0400 Subject: [PATCH] Change default for HF on non-breakmodel models to use GPU instead of CPU --- modeling/inference_models/hf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modeling/inference_models/hf.py b/modeling/inference_models/hf.py index 032b8ec3..ee585321 100644 --- a/modeling/inference_models/hf.py +++ b/modeling/inference_models/hf.py @@ -133,7 +133,7 @@ class HFInferenceModel(InferenceModel): "unit": "bool", "label": "Use GPU", "id": "use_gpu", - "default": False, + "default": True, "tooltip": "Whether or not to use the GPU", "menu_path": "Layers", "extra_classes": "",