make style

cb8a3dbe · Patrick von Platen · bcd6f3f9 · cb8a3dbe
Commit cb8a3dbe authored Jan 04, 2023 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

examples/dreambooth/train_dreambooth_flax.py examples/dreambooth/train_dreambooth_flax.py +1 -3

No files found.
--- a/examples/dreambooth/train_dreambooth_flax.py
+++ b/examples/dreambooth/train_dreambooth_flax.py
@@ -629,9 +629,7 @@ def main():
    def checkpoint(step=None):
        # Create the pipeline using the trained modules and save it.
-        scheduler, _ = FlaxPNDMScheduler.from_pretrained(
+        scheduler, _ = FlaxPNDMScheduler.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="scheduler")
-                "CompVis/stable-diffusion-v1-4", subfolder="scheduler"
-        )
        safety_checker = FlaxStableDiffusionSafetyChecker.from_pretrained(
            "CompVis/stable-diffusion-safety-checker", from_pt=True
        )