apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-11 16:52:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update

Browse source
This commit is contained in:
stellahsr 2024-01-11 19:29:49 +08:00
parent 9e0b9745be
commit e56caa6f5e
2 changed files with 42 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/prompts/ml_engineer.py
View file

@ -58,6 +58,7 @@ Please assign a task type to each task in the list below from the given categori
- **data_preprocess**: Only for changing value inplace.
- **model_train**: Only for training model.
- **model_evaluate**: Only for evaluating model.
- **stable_diffusion**: Related to text2image, image2image using stable diffusion model.
- **other**: Any tasks that do not fit into the previous categories, such as visualization, summarizing findings, etc.
"""
@ -326,4 +327,5 @@ TASK_MODULE_MAP = {
"data_preprocess": "metagpt.tools.functions.libs.data_preprocess",
"feature_engineering": "metagpt.tools.functions.libs.feature_engineering",
"udf": "metagpt.tools.functions.libs.udf",
"stable_diffusion": "metagpt.tools.sd_engine",
}