tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

change deprecated modal Stub to App (#2038)

Browse Source
This commit is contained in:
Wing Lian
2024-11-11 15:10:34 -05:00
committed by GitHub
parent dc8f9059f7
commit 3931a42763
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cicd/multigpu.py
View File

@@ -10,7 +10,7 @@ import tempfile
import jinja2 import jinja2
import modal import modal
from jinja2 import select_autoescape from jinja2 import select_autoescape
from modal import Image, Stub from modal import App, Image
cicd_path = pathlib.Path(__file__).parent.resolve() cicd_path = pathlib.Path(__file__).parent.resolve()
@@ -46,7 +46,7 @@ cicd_image = (
.pip_install("fastapi==0.110.0", "pydantic==2.6.3") .pip_install("fastapi==0.110.0", "pydantic==2.6.3")
) )
stub = Stub("Axolotl CI/CD", secrets=[]) app = App("Axolotl CI/CD", secrets=[])
N_GPUS = int(os.environ.get("N_GPUS", 2)) N_GPUS = int(os.environ.get("N_GPUS", 2))
@@ -61,7 +61,7 @@ def run_cmd(cmd: str, run_folder: str):
exit(exit_code) # pylint: disable=consider-using-sys-exit exit(exit_code) # pylint: disable=consider-using-sys-exit
@stub.function( @app.function(
image=cicd_image, image=cicd_image,
gpu=GPU_CONFIG, gpu=GPU_CONFIG,
timeout=60 * 60, timeout=60 * 60,
@@ -72,6 +72,6 @@ def cicd_pytest():
run_cmd("./cicd/multigpu.sh", "/workspace/axolotl") run_cmd("./cicd/multigpu.sh", "/workspace/axolotl")
@stub.local_entrypoint() @app.local_entrypoint()
def main(): def main():
cicd_pytest.remote() cicd_pytest.remote()

8
cicd/tests.py
View File

@@ -10,7 +10,7 @@ import tempfile
import jinja2 import jinja2
import modal import modal
from jinja2 import select_autoescape from jinja2 import select_autoescape
from modal import Image, Stub from modal import App, Image
cicd_path = pathlib.Path(__file__).parent.resolve() cicd_path = pathlib.Path(__file__).parent.resolve()
@@ -47,7 +47,7 @@ cicd_image = (
.pip_install("fastapi==0.110.0", "pydantic==2.6.3") .pip_install("fastapi==0.110.0", "pydantic==2.6.3")
) )
stub = Stub("Axolotl CI/CD", secrets=[]) app = App("Axolotl CI/CD", secrets=[])
N_GPUS = int(os.environ.get("N_GPUS", 1)) N_GPUS = int(os.environ.get("N_GPUS", 1))
@@ -62,7 +62,7 @@ def run_cmd(cmd: str, run_folder: str):
exit(exit_code) # pylint: disable=consider-using-sys-exit exit(exit_code) # pylint: disable=consider-using-sys-exit
@stub.function( @app.function(
image=cicd_image, image=cicd_image,
gpu=GPU_CONFIG, gpu=GPU_CONFIG,
timeout=60 * 60, timeout=60 * 60,
@@ -73,6 +73,6 @@ def cicd_pytest():
run_cmd("./cicd/cicd.sh", "/workspace/axolotl") run_cmd("./cicd/cicd.sh", "/workspace/axolotl")
@stub.local_entrypoint() @app.local_entrypoint()
def main(): def main():
cicd_pytest.remote() cicd_pytest.remote()