fix linting

drhead · Jun 22, 2024 · cff1e06 · cff1e06
1 parent 48039c3
commit cff1e06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/training_utils.py b/src/diffusers/training_utils.py
@@ -350,7 +350,7 @@ def __init__(
         self.model_config = model_config
 
     @classmethod
-    def from_pretrained(cls, path, model_cls, foreach = False) -> "EMAModel":
+    def from_pretrained(cls, path, model_cls, foreach=False) -> "EMAModel":
         _, ema_kwargs = model_cls.load_config(path, return_unused_kwargs=True)
         model = model_cls.from_pretrained(path)