Skip to content

Commit baea141

Browse files
committed
make
1 parent 5af8c7f commit baea141

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -848,7 +848,9 @@ def load_module(name, value):
848848
continue
849849
class_name = passed_class_obj[key].__class__.__name__
850850
class_name = class_name[4:] if class_name.startswith("Flax") else class_name
851-
expected_class_name = expected_class_name[4:] if expected_class_name.startswith("Flax") else expected_class_name
851+
expected_class_name = (
852+
expected_class_name[4:] if expected_class_name.startswith("Flax") else expected_class_name
853+
)
852854
if key == "scheduler" and scheduler_types is not None and class_name not in scheduler_types:
853855
raise ValueError(f"Expected {scheduler_types} for {key}, got {class_name}.")
854856
elif key != "scheduler" and class_name != expected_class_name:

0 commit comments

Comments
 (0)