Skip to content

Commit fb5d51a

Browse files
authored
Merge pull request #385 from anyangml2nd/fix/change-machine-type
Fix: use explicit machine type
2 parents 1dd488e + 18648aa commit fb5d51a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lambench/tasks/base_task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class BaseTask(BaseModel):
3131
model_config = ConfigDict(extra="allow")
3232
workdir: Path = Path(tempfile.gettempdir()) / "lambench"
3333
record_type: ClassVar = BaseRecord
34-
machine_type: str = "1 * NVIDIA V100_32g"
34+
machine_type: str = "c12_m92_1 * NVIDIA V100"
3535

3636
def exist(self, model_name: str) -> bool:
3737
num_records = self.record_type.count(

0 commit comments

Comments
 (0)