github/ChatDev
1
0
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2026-08-01 19:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix multi-model support

Browse Source
This commit is contained in:
thinkwee 2024-11-13 18:33:58 +08:00
parent 8f5c68ab1c
commit f44e91dd16
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graph.py
View File

@ -264,7 +264,7 @@ class Graph:
if len(line) == 1: # In case there is only one layer if len(line) == 1: # In case there is only one layer
for node_id in range(line[0][0][0], line[0][0][1] + 1): for node_id in range(line[0][0][0], line[0][0][1] + 1):
if node_id not in self.nodes: if node_id not in self.nodes:
self.add_node(Node(node_id)) self.add_node(Node(node_id), model=self.config.get("Model"))
for i in range(len(line) - 1): for i in range(len(line) - 1):
from_node_list = line[i] from_node_list = line[i]
to_node_list = line[i + 1] to_node_list = line[i + 1]
@ -273,9 +273,9 @@ class Graph:
for to_node_tuple in to_node_list: for to_node_tuple in to_node_list:
for to_node_id in range(to_node_tuple[0], to_node_tuple[1] + 1): for to_node_id in range(to_node_tuple[0], to_node_tuple[1] + 1):
if from_node_id not in self.nodes: if from_node_id not in self.nodes:
self.add_node(Node(from_node_id)) self.add_node(Node(from_node_id), model=self.config.get("Model"))
if to_node_id not in self.nodes: if to_node_id not in self.nodes:
self.add_node(Node(to_node_id)) self.add_node(Node(to_node_id), model=self.config.get("Model"))
self.add_edge(from_node_id, to_node_id) self.add_edge(from_node_id, to_node_id)
self.input_layer = self.get_input_layer() self.input_layer = self.get_input_layer()
self.output_layer = self.get_output_layer() self.output_layer = self.get_output_layer()