github/ChatDev
1
0
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2026-04-25 11:18:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix multi-model support

Browse Source
This commit is contained in:
thinkwee 2024-11-26 10:09:55 +08:00
parent f44e91dd16
commit bd59d89c7f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
graph.py
View File

@ -273,9 +273,9 @@ class Graph:
for to_node_tuple in to_node_list:
for to_node_id in range(to_node_tuple[0], to_node_tuple[1] + 1):
if from_node_id not in self.nodes:
self.add_node(Node(from_node_id), model=self.config.get("Model"))
self.add_node(Node(from_node_id, model=self.config.get("Model")))
if to_node_id not in self.nodes:
self.add_node(Node(to_node_id), model=self.config.get("Model"))
self.add_node(Node(to_node_id, model=self.config.get("Model")))
self.add_edge(from_node_id, to_node_id)
self.input_layer = self.get_input_layer()
self.output_layer = self.get_output_layer()