sidphbot commited on
Commit
3eee270
1 Parent(s): 1e609d1

Update src/Surveyor.py

Browse files
Files changed (1) hide show
  1. src/Surveyor.py +3 -3
src/Surveyor.py CHANGED
@@ -87,9 +87,9 @@ class Surveyor:
87
  spacy.require_gpu()
88
 
89
  self.high_gpu = high_gpu
90
- self.DEFAULTS = DEFAULTS_CPU_COMPAT
91
- if self.high_gpu:
92
- self.DEFAULTS = DEFAULTS_HIGH_GPU
93
 
94
  if not kw_model_name:
95
  kw_model_name = self.DEFAULTS["kw_model_name"]
 
87
  spacy.require_gpu()
88
 
89
  self.high_gpu = high_gpu
90
+ #self.DEFAULTS = DEFAULTS_CPU_COMPAT
91
+ #if self.high_gpu:
92
+ self.DEFAULTS = DEFAULTS_HIGH_GPU
93
 
94
  if not kw_model_name:
95
  kw_model_name = self.DEFAULTS["kw_model_name"]