You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
from swarms.agents.base import agent
|
|
|
|
|
from swarms.structs.nonlinear_worfklow import NonLinearWorkflow, Task
|
|
|
|
|
|
|
|
|
|
prompt = "develop a feedforward network in pytorch"
|
|
|
|
|
prompt2 = "Develop a self attention using pytorch"
|
|
|
|
|
|
|
|
|
|
task1 = Task("task1", prompt)
|
|
|
|
|
task2 = Task("task2", prompt2, parents=[task1])
|
|
|
|
|
|
|
|
|
|
# add tasks to workflow
|
|
|
|
|
workflow = NonLinearWorkflow(agent)
|
|
|
|
|
|
|
|
|
|
# add tasks to tree
|
|
|
|
|
workflow.add(task1)
|
|
|
|
|
workflow.add(task2)
|
|
|
|
|
|
|
|
|
|
# run
|
|
|
|
|
workflow.run()
|
