Lightning-Universe / stable-diffusion-deploy

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.

Home Page:https://lightning.ai/muse

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

unable to run the app on cloud

aniketmaurya opened this issue Β· comments

πŸ› Bug

[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):
  • Python version:
  • CUDA/cuDNN version:
  • GPU models and configuration:
  • Any other relevant information:

Additional context

@aniketmaurya let's close this?

yes, closing since the issue is fixed and I am able to use Muse local and cloud