From 85665a63ec95bd3885b887488d0004011d4cc666 Mon Sep 17 00:00:00 2001 From: Charles Frye Date: Thu, 3 Oct 2024 12:02:17 -0400 Subject: [PATCH] provide a default value for compile so @build runs (#912) --- 06_gpu_and_ml/stable_diffusion/flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/06_gpu_and_ml/stable_diffusion/flux.py b/06_gpu_and_ml/stable_diffusion/flux.py index ecb2bd5f9..4af46bedb 100644 --- a/06_gpu_and_ml/stable_diffusion/flux.py +++ b/06_gpu_and_ml/stable_diffusion/flux.py @@ -120,7 +120,7 @@ ) class Model: compile: int = ( # see section on torch.compile below for details - modal.parameter() + modal.parameter(default=0) ) def setup_model(self):