Xu Xuenan commited on
Commit
18d0cba
·
1 Parent(s): 4017e9d

Update mm_story_agent/__init__.py

Browse files
Files changed (1) hide show
  1. mm_story_agent/__init__.py +1 -11
mm_story_agent/__init__.py CHANGED
@@ -59,7 +59,7 @@ class MMStoryAgent:
59
  return_dict = mp.Manager().dict()
60
 
61
  for modality in self.modalities:
62
- p = mp.Process(target=self.call_modality_agent, args=(agents[modality], pages, story_dir / modality, return_dict))
63
  processes.append(p)
64
  p.start()
65
 
@@ -93,13 +93,3 @@ class MMStoryAgent:
93
  pages = self.write_story(config)
94
  images = self.generate_modality_assets(config, pages)
95
  self.compose_storytelling_video(config, pages)
96
-
97
-
98
- if __name__ == "__main__":
99
-
100
- from arg_parser import parse_yaml_and_cmd
101
-
102
- config = parse_yaml_and_cmd()
103
- mm_story_agent = MMStoryAgent()
104
-
105
- mm_story_agent.call(config)
 
59
  return_dict = mp.Manager().dict()
60
 
61
  for modality in self.modalities:
62
+ p = mp.Process(target=self.call_modality_agent, args=(agents[modality], pages, story_dir / modality, return_dict), daemon=False)
63
  processes.append(p)
64
  p.start()
65
 
 
93
  pages = self.write_story(config)
94
  images = self.generate_modality_assets(config, pages)
95
  self.compose_storytelling_video(config, pages)