Skip to content

Commit bccda52

Browse files
committed
update
1 parent f36ba9f commit bccda52

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,8 @@ def __init__(self):
367367
def initialize_hook(self, module):
368368
def make_execution_order_update_callback(current_name, current_submodule):
369369
def callback():
370-
logger.debug(f"Adding {current_name} to the execution order")
370+
if not torch.compiler.is_compiling():
371+
logger.debug(f"Adding {current_name} to the execution order")
371372
self.execution_order.append((current_name, current_submodule))
372373

373374
return callback

0 commit comments

Comments
 (0)