Spaces:
Running
on
L4
Running
on
L4
Xu Xuenan
commited on
Commit
·
18d0cba
1
Parent(s):
4017e9d
Update mm_story_agent/__init__.py
Browse files- 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|