| 327 | 'Dockerfile', '.') |
| 328 | |
| 329 | def push(self): |
| 330 | ret = self.run_cmd('docker', 'push', self.image()) |
| 331 | if ret != 0: |
| 332 | return ret |
| 333 | image_tag2 = ( |
| 334 | f'{docker_registry}:ubuntu{self.args.ubuntu_version}-{self.args.python_tag}-' |
| 335 | f'torch{self.args.torch_version}-{self.args.modelscope_version}-{formatted_time}-test' |
| 336 | ) |
| 337 | ret = self.run_cmd('docker', 'tag', self.image(), image_tag2) |
| 338 | if ret != 0: |
| 339 | return ret |
| 340 | return self.run_cmd('docker', 'push', image_tag2) |
| 341 | |
| 342 | |
| 343 | class StableGPUImageBuilder(Builder): |