update to latest version of diffusers

requirements-lock.txt

@@ -1,183 +1,110 @@
 absl-py==1.2.0
 accelerate==0.12.0
-aiobotocore==2.3.4
 aiohttp==3.8.1
-aioitertools==0.10.0
 aiosignal==1.2.0
-analytics-python==1.4.0
-anyio==3.6.1
 appdirs==1.4.4
 argon2-cffi==21.3.0
 argon2-cffi-bindings==21.2.0
+asttokens==2.0.8
 async-timeout==4.0.2
-attrs==21.4.0
+attrs==22.1.0
 audioread==3.0.0
 backcall==0.2.0
-backoff==1.10.0
-bcrypt==3.2.2
 beautifulsoup4==4.11.1
-bertviz==1.4.0
-black==22.6.0
 bleach==5.0.1
-boto3==1.21.21
-botocore==1.24.21
 cachetools==5.2.0
-captum==0.5.0
 certifi==2022.6.15
 cffi==1.15.1
-cfgv==3.3.1
-charset-normalizer==2.1.0
-click==8.1.3
-cloudpickle==2.1.0
-colossalai==0.1.8
-commonmark==0.9.1
-cryptography==37.0.4
-cycler==0.11.0
+charset-normalizer==2.1.1
 datasets==2.4.0
-debugpy==1.6.2
+debugpy==1.6.3
 decorator==5.1.1
-deepspeed==0.7.0
 defusedxml==0.7.1
-diffusers==0.2.2
+diffusers==0.2.4
 dill==0.3.5.1
-distlib==0.3.5
 entrypoints==0.4
-fabric==2.7.1
-fastapi==0.79.0
+executing==0.10.0
 fastjsonschema==2.16.1
-ffmpy==0.3.0
 filelock==3.8.0
-fonttools==4.34.4
 frozenlist==1.3.1
 fsspec==2022.7.1
-google-auth==2.10.0
+ftfy==6.1.1
+google-auth==2.11.0
 google-auth-oauthlib==0.4.6
-google-pasta==0.2.0
-gradio==3.1.4
 grpcio==1.47.0
-h11==0.12.0
-hjson==3.1.0
-httpcore==0.15.0
-httpx==0.23.0
-huggingface-hub==0.8.1
-identify==2.5.3
+huggingface-hub==0.9.0
 idna==3.3
 importlib-metadata==4.12.0
-invoke==1.7.1
 ipykernel==6.15.1
-ipython==7.34.0
+ipython==8.4.0
 ipython-genutils==0.2.0
 ipywidgets==7.7.1
 jedi==0.18.1
 Jinja2==3.1.2
-jmespath==1.0.1
 joblib==1.1.0
-jsonschema==4.9.1
+jsonschema==4.14.0
 jupyter-client==7.3.4
+jupyter-console==6.4.4
 jupyter-core==4.11.1
 jupyterlab-pygments==0.2.2
-jupyterlab-widgets==1.1.1
-kiwisolver==1.4.4
+jupyterlab-widgets==3.0.2
 librosa==0.9.2
-linkify-it-py==1.0.3
 llvmlite==0.39.0
 lxml==4.9.1
 Markdown==3.4.1
-markdown-it-py==2.1.0
 MarkupSafe==2.1.1
-matplotlib==3.5.3
-matplotlib-inline==0.1.3
-mdit-py-plugins==0.3.0
-mdurl==0.1.2
-mistune==0.8.4
-monotonic==1.6
-more-itertools==8.14.0
+matplotlib-inline==0.1.6
+mistune==2.0.4
 multidict==6.0.2
 multiprocess==0.70.13
-munkres==1.1.4
-mypy-extensions==0.4.3
-nbclient==0.6.6
-nbconvert==6.5.3
+nbclient==0.6.7
+nbconvert==7.0.0
 nbformat==5.4.0
 nest-asyncio==1.5.5
-networkx==2.8.5
-ninja==1.10.2.3
-nlp==0.4.0
-nltk==3.7
-nodeenv==1.7.0
 notebook==6.4.12
 numba==0.56.0
 numpy==1.22.4
 oauthlib==3.2.0
-orjson==3.7.12
 packaging==21.3
 pandas==1.4.3
 pandocfilters==1.5.0
-paramiko==2.11.0
 parso==0.8.3
-pathlib2==2.3.7.post1
-pathos==0.2.9
-pathspec==0.9.0
 pexpect==4.8.0
 pickleshare==0.7.5
 Pillow==9.2.0
-platformdirs==2.5.2
-pluggy==0.13.1
 pooch==1.6.0
-pox==0.3.1
-ppft==1.7.6.5
-pre-commit==2.20.0
 prometheus-client==0.14.1
 prompt-toolkit==3.0.30
 protobuf==3.19.4
-protobuf3-to-dict==0.1.5
 psutil==5.9.1
 ptyprocess==0.7.0
-py==1.11.0
-py-cpuinfo==8.0.0
+pure-eval==0.2.2
 pyarrow==9.0.0
 pyasn1==0.4.8
 pyasn1-modules==0.2.8
 pycparser==2.21
-pycryptodome==3.15.0
-pydantic==1.9.2
-pydub==0.25.1
-Pygments==2.12.0
-PyNaCl==1.5.0
+Pygments==2.13.0
 pyparsing==3.0.9
 pyrsistent==0.18.1
-pytest==5.4.3
 python-dateutil==2.8.2
-python-dotenv==0.20.0
-python-multipart==0.0.5
 pytz==2022.2.1
 PyYAML==6.0
 pyzmq==23.2.1
-regex==2022.7.25
+qtconsole==5.3.1
+QtPy==2.2.0
+regex==2022.8.17
 requests==2.28.1
 requests-oauthlib==1.3.1
 resampy==0.4.0
 responses==0.18.0
-rfc3986==1.5.0
-rich==12.5.1
 rsa==4.9
-s3fs==2022.7.1
-s3transfer==0.5.2
-sagemaker==2.103.0
 scikit-learn==1.1.2
 scipy==1.9.0
-seaborn==0.11.2
 Send2Trash==1.8.0
-sentencepiece==0.1.97
-shap==0.41.0
 six==1.16.0
-slicer==0.0.7
-smdebug-rulesconfig==1.0.1
-sniffio==1.2.0
-snorkel==0.9.9
 SoundFile==0.10.3.post1
 soupsieve==2.3.2.post1
-starlette==0.19.1
+stack-data==0.4.0
 tensorboard==2.10.0
 tensorboard-data-server==0.6.1
 tensorboard-plugin-wit==1.8.1
@@ -186,25 +113,18 @@ threadpoolctl==3.1.0
 tinycss2==1.1.1
 tokenizers==0.12.1
 toml==0.10.2
-tomli==2.0.1
-torch==1.12.1+cu116
-torchaudio==0.12.1+cu116
-torchvision==0.13.1+cu116
+torch==1.12.1
+torchvision==0.13.1
 tornado==6.2
 tqdm==4.64.0
 traitlets==5.3.0
 transformers==4.21.1
-transformers-interpret==0.7.5
 typing_extensions==4.3.0
-uc-micro-py==1.0.1
-urllib3==1.26.11
-uvicorn==0.18.2
-virtualenv==20.16.3
+urllib3==1.26.12
 wcwidth==0.2.5
 webencodings==0.5.1
 Werkzeug==2.2.2
 widgetsnbextension==3.6.1
-wrapt==1.14.1
 xxhash==3.0.0
 yapf==0.32.0
 yarl==1.8.1

src/train_unconditional.py

@@ -206,7 +206,22 @@ def main(args):
 
         # Generate sample images for visual inspection
         if accelerator.is_main_process:
-            if epoch % args.save_images_epochs == 0 or epoch == args.num_epochs - 1:
+            if epoch % args.save_model_epochs == 0 or epoch == args.num_epochs - 1:
+                # save the model
+                if args.push_to_hub:
+                    try:
+                        push_to_hub(
+                            args,
+                            pipeline,
+                            repo,
+                            commit_message=f"Epoch {epoch}",
+                            blocking=False,
+                        )
+                    except NameError:  # current version of diffusers has a little bug
+                        pass
+                else:
+                    pipeline.save_pretrained(output_dir)
+
                 pipeline = DDPMPipeline(
                     unet=accelerator.unwrap_model(
                         ema_model.averaged_model if args.use_ema else model
@@ -237,22 +252,6 @@ def main(args):
                         epoch,
                         sample_rate=mel.get_sample_rate(),
                     )
-
-            if epoch % args.save_model_epochs == 0 or epoch == args.num_epochs - 1:
-                # save the model
-                if args.push_to_hub:
-                    try:
-                        push_to_hub(
-                            args,
-                            pipeline,
-                            repo,
-                            commit_message=f"Epoch {epoch}",
-                            blocking=False,
-                        )
-                    except NameError:  # current version of diffusers has a little bug
-                        pass
-                else:
-                    pipeline.save_pretrained(output_dir)
         accelerator.wait_for_everyone()
 
     accelerator.end_training()