|
|
|
@@ -29,6 +29,10 @@ class DialogModelingPreprocessor(Preprocessor): |
|
|
|
self.model_dir: str = model_dir |
|
|
|
self.config = Config.from_file( |
|
|
|
os.path.join(self.model_dir, ModelFile.CONFIGURATION)) |
|
|
|
|
|
|
|
import torch |
|
|
|
self.config.use_gpu = self.config.use_gpu and torch.cuda.is_available() |
|
|
|
|
|
|
|
self.text_field = MultiWOZBPETextField( |
|
|
|
self.model_dir, config=self.config) |
|
|
|
|
|
|
|
|