Skip to content

Commit 2ba8d4d

Browse files
committed
add torch.compiler.disable to progress_bar to keep it in sync with the base
1 parent d6ff273 commit 2ba8d4d

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

src/diffusers/pipelines/marigold/pipeline_marigold_depth.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,6 +321,7 @@ def check_inputs(
321321

322322
return num_images
323323

324+
@torch.compiler.disable
324325
def progress_bar(self, iterable=None, total=None, desc=None, leave=True):
325326
if not hasattr(self, "_progress_bar_config"):
326327
self._progress_bar_config = {}

src/diffusers/pipelines/marigold/pipeline_marigold_intrinsics.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,7 @@ def check_inputs(
329329

330330
return num_images
331331

332+
@torch.compiler.disable
332333
def progress_bar(self, iterable=None, total=None, desc=None, leave=True):
333334
if not hasattr(self, "_progress_bar_config"):
334335
self._progress_bar_config = {}

src/diffusers/pipelines/marigold/pipeline_marigold_normals.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -306,6 +306,7 @@ def check_inputs(
306306

307307
return num_images
308308

309+
@torch.compiler.disable
309310
def progress_bar(self, iterable=None, total=None, desc=None, leave=True):
310311
if not hasattr(self, "_progress_bar_config"):
311312
self._progress_bar_config = {}

0 commit comments

Comments
 (0)