kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make style

Browse Source
This commit is contained in:
Patrick von Platen
2023-01-04 21:50:17 +00:00
parent bcd6f3f9ce
commit cb8a3dbe34
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/dreambooth/train_dreambooth_flax.py
+1 -3
View File
@@ -629,9 +629,7 @@ def main():
def checkpoint(step=None): def checkpoint(step=None):
# Create the pipeline using the trained modules and save it. # 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( safety_checker = FlaxStableDiffusionSafetyChecker.from_pretrained(
"CompVis/stable-diffusion-safety-checker", from_pt=True "CompVis/stable-diffusion-safety-checker", from_pt=True
) )