We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f36ba9f commit bccda52Copy full SHA for bccda52
src/diffusers/hooks/group_offloading.py
@@ -367,7 +367,8 @@ def __init__(self):
367
def initialize_hook(self, module):
368
def make_execution_order_update_callback(current_name, current_submodule):
369
def callback():
370
- logger.debug(f"Adding {current_name} to the execution order")
+ if not torch.compiler.is_compiling():
371
+ logger.debug(f"Adding {current_name} to the execution order")
372
self.execution_order.append((current_name, current_submodule))
373
374
return callback
0 commit comments