Fix issues for a 'DistributedDataParallel' object (#3787)

Fix error to address the `AttributeError: 'DistributedDataParallel' object has no attribute 'tokenize_query'`
facebookresearch · Jul 15, 2021 · de5a42d · de5a42d
1 parent ef1511f
commit de5a42d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/parlai/agents/rag/rag.py b/parlai/agents/rag/rag.py
@@ -509,7 +509,10 @@ def _set_query_vec(self, observation: Message) -> Message:
             return observation with query vec.
         """
         query_str = observation[self._query_key]
-        observation['query_vec'] = self.model.tokenize_query(query_str)
+        if hasattr(self.model, 'module'):
+            observation['query_vec'] = self.model.module.tokenize_query(query_str)
+        else:
+            observation['query_vec'] = self.model.tokenize_query(query_str)
         return observation
 
     def _set_input_turn_cnt_vec(self, observation: Message) -> Message: