diff --git a/06_gpu_and_ml/gpu_fallbacks.py b/06_gpu_and_ml/gpu_fallbacks.py index 7647a3bd3..fb3c82c51 100644 --- a/06_gpu_and_ml/gpu_fallbacks.py +++ b/06_gpu_and_ml/gpu_fallbacks.py @@ -1,9 +1,3 @@ -import subprocess - -import modal - -app = modal.App("example-gpu-fallbacks") - # # Set "fallback" GPUs # # GPU availabilities on Modal can fluctuate, especially for @@ -16,6 +10,12 @@ # # The code below demonstrates the usage of the `gpu` parameter with a list of GPUs. +import subprocess + +import modal + +app = modal.App("example-gpu-fallbacks") + @app.function( gpu=["h100", "a100", "any"], # "any" means any of L4, A10, or T4