Hugging Face's logo

Spaces:
ashu316
/
tgn-playground
Running

App Files Community
ashu316 commited on
Commit
5f8ff92
·
verified ·
1 Parent(s): 4e3dad4

Update modules/memory_updater.py

Browse files
Files changed (1) hide show
  1. modules/memory_updater.py +2 -1
modules/memory_updater.py CHANGED
@@ -28,7 +28,8 @@ class SequenceMemoryUpdater(MemoryUpdater):
28
 
29
  updated_memory = self.memory_updater(unique_messages, memory)
30
 
31
- self.memory.set_memory(unique_node_ids, updated_memory)
 
32
 
33
  def get_updated_memory(self, unique_node_ids, unique_messages, timestamps):
34
  if len(unique_node_ids) <= 0:
 
28
 
29
  updated_memory = self.memory_updater(unique_messages, memory)
30
 
31
+ with torch.no_grad():
32
+ self.memory.set_memory(unique_node_ids, updated_memory)
33
 
34
  def get_updated_memory(self, unique_node_ids, unique_messages, timestamps):
35
  if len(unique_node_ids) <= 0: