add t.stop()

Author: casteryh

src/forge/actors/generator.py

@@ -305,6 +305,8 @@ def split_keys(keys):
         for sd in sub_state_dicts:
             state_dict.update(sd)
 
+        t.stop()
+
         return state_dict
 
     @endpoint