mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-03 18:27:43 +00:00
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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user