From 0045260af78d54bf1dfb49e8dbac750de69e7594 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 31 Mar 2026 09:27:54 -0600 Subject: [PATCH] Fix issue where compile true did not actually compile the model --- jobs/process/BaseSDTrainProcess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index dae17596..0ca4680c 100644 --- a/jobs/process/BaseSDTrainProcess.py +++ b/jobs/process/BaseSDTrainProcess.py @@ -1618,7 +1618,7 @@ class BaseSDTrainProcess(BaseTrainProcess): # compile the model if needed if self.model_config.compile: try: - torch.compile(self.sd.unet, dynamic=True, fullgraph=True, mode='max-autotune') + self.sd.unet = torch.compile(self.sd.unet, dynamic=True, fullgraph=True, mode='max-autotune') except Exception as e: print_acc(f"Failed to compile model: {e}") print_acc("Continuing without compilation")