Fixed issue where use_gpu=False still uses GPU.

owo · owo · commit b496501590ee · 2023-04-14T10:20:56.000+04:00
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

Original file line number	Diff line number	Diff line change
`@@ -254,7 +254,7 @@ def __init__(self, model_path, analyzer,`
`254`	`254`	`scorer='uniform', tie_breaker='tag', use_gpu=True,`
`255`	`255`	`batch_size=32, ranking_cache=None, ranking_cache_size=100000):`
`256`	`256`	`self._model = {`
`257`		`- 'unfactored': _BERTFeatureTagger(model_path)`
	`257`	`+ 'unfactored': _BERTFeatureTagger(model_path, use_gpu=use_gpu)`
`258`	`258`	`}`
`259`	`259`	`self._analyzer = analyzer`
`260`	`260`	`self._features = features`