Fixed issue where use_gpu=False still uses GPU.

CAMeL-Lab · Apr 14, 2023 · b496501 · b496501
1 parent 288202d
commit b496501
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/camel_tools/disambig/bert/unfactored.py b/camel_tools/disambig/bert/unfactored.py
@@ -254,7 +254,7 @@ def __init__(self, model_path, analyzer,
                  scorer='uniform', tie_breaker='tag', use_gpu=True,
                  batch_size=32, ranking_cache=None, ranking_cache_size=100000):
         self._model = {
-            'unfactored': _BERTFeatureTagger(model_path)
+            'unfactored': _BERTFeatureTagger(model_path, use_gpu=use_gpu)
         }
         self._analyzer = analyzer
         self._features = features