removed the offset (facebookresearch#4331)

josephwarrick · Jan 27, 2022 · d6b2fb1 · d6b2fb1
1 parent 6fbd1a3
commit d6b2fb1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/projects/blenderbot2/agents/sub_modules.py b/projects/blenderbot2/agents/sub_modules.py
@@ -313,12 +313,10 @@ def generate_memories(
         """
         assert self.agent_dict is not None
         memories = []
-        offset = 0
-        for idx, i in enumerate(input):
+        for idx, input_i in enumerate(input):
             if num_inputs[idx] == 0:
                 continue
-            context_lines_vec = i[offset : offset + num_inputs[idx]]
-            offset += num_inputs[idx]
+            context_lines_vec = input_i[: num_inputs[idx]]
             context_lines = [
                 self.agent_dict.vec2txt(self.clean_input(j)) for j in context_lines_vec
             ]