Learn to serve Stable Diffusion models on cloud infrastructure at scale. This Lightning App shows load-balancing, orchestrating, pre-provisioning, dynamic batching, GPU-inference, micro-services working together via the Lightning Apps framework.
[flow] 2022-11-02T10:01:06.189Z INFO: Waiting for works to stop...
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.394Z /home/zeus/.local/lib/python3.8/site-packages/pkg_resources/__init__.py:116: PkgResourcesDeprecationWarning: 0.23ubuntu1 is an invalid version and will not be supported in a future release
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.394Z warnings.warn(
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.394Z /home/zeus/.local/lib/python3.8/site-packages/pkg_resources/__init__.py:116: PkgResourcesDeprecationWarning: 0.23ubuntu1 is an invalid version and will not be supported in a future release
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.394Z warnings.warn(
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z Traceback (most recent call last):
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z sys.exit(main())
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 1130, in __call__
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z return self.main(*args, **kwargs)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 1055, in main
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z rv = self.invoke(ctx)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 1657, in invoke
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z return _process_result(sub_ctx.command.invoke(sub_ctx))
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 1657, in invoke
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z return _process_result(sub_ctx.command.invoke(sub_ctx))
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 1404, in invoke
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z return ctx.invoke(self.callback, **ctx.params)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/click/core.py", line 760, in invoke
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z return __callback(*args, **kwargs)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning_launcher/cli/__main__.py", line 87, in run_work
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z run_lightning_work(
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning_launcher/utils.py", line 50, in wrapper
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z res = func(*args, **kwargs)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning_launcher/utils.py", line 75, in wrapper
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z res = func(*args, **kwargs)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning_launcher/launcher.py", line 161, in run_lightning_work
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z WorkRunner(
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning/app/utilities/proxies.py", line 336, in __call__
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z raise e
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning/app/utilities/proxies.py", line 319, in __call__
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z self.run_once()
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning/app/utilities/proxies.py", line 457, in run_once
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z self.work.on_exception(e)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning/app/core/work.py", line 550, in on_exception
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z raise exception
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/lightning/app/utilities/proxies.py", line 422, in run_once
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z ret = work_run(*args, **kwargs)
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/muse/components/safety_checker_embedding.py", line 28, in run
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z from flash import Trainer
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/flash/__init__.py", line 22, in <module>
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z from flash.core.data.callback import FlashCallback
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z File "/home/zeus/.local/lib/python3.8/site-packages/flash/core/data/callback.py", line 4, in <module>
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z from pytorch_lightning.callbacks import Callback
[root.safety_checker_embedding_work] 2022-11-02T10:01:05.397Z ModuleNotFoundError: No module named 'pytorch_lightning'
To Reproduce
Steps to reproduce the behavior:
clone the main branch and launch in cloud
Code sample
Expected behavior
Environment
PyTorch Version (e.g., 1.0):
OS (e.g., Linux):
How you installed PyTorch (conda, pip, source):
Build command you used (if compiling from source):
🐛 Bug
To Reproduce
Steps to reproduce the behavior:
clone the main branch and launch in cloud
Code sample
Expected behavior
Environment
conda
,pip
, source):Additional context