feat: dump model (#776)

* feat: add model dump flag and multi-evaluator support

* tmp code

* refactor: update evaluator feedback and FBWorkspace types

* feat: add get_clear_ws_cmd and CPU count in Docker environment

* feat: Add model dump check level and enhance evaluator functionality

fix data type bug

* fix: Ensure required files exist before model dump evaluation

* refactor: streamline prompt and file checks in model dump evaluation

* fix: add assertions and reorder file reads in model dump evaluator

* feat: remove EDA part from evaluation output

* docs: update dump_model guidelines and eval prompt to include template

* style: reformat multiline dicts and lists in conf and eval files

* fix: add DOTALL flag to EDA removal regex
This commit is contained in:
you-n-g
2025-04-09 23:24:12 +08:00
committed by GitHub
parent 835da5b918
commit bbc591b401
13 changed files with 271 additions and 25 deletions
+2
View File
@@ -397,6 +397,7 @@ class DockerConf(EnvConf):
shm_size: str | None = None
enable_gpu: bool = True # because we will automatically disable GPU if not available. So we enable it by default.
mem_limit: str | None = "48g" # Add memory limit attribute
cpu_count: int | None = None # Add CPU limit attribute
running_timeout_period: int = 3600 # 1 hour
@@ -633,6 +634,7 @@ class DockerEnv(Env[DockerConf]):
network=self.conf.network,
shm_size=self.conf.shm_size,
mem_limit=self.conf.mem_limit, # Set memory limit
cpu_count=self.conf.cpu_count, # Set CPU limit
**self._gpu_kwargs(client),
)
logs = container.logs(stream=True)