search

Engineer-of-Stuff / stable-diffusion-paperspace

Jupyter notebooks for Paperspace.
The Unlicense
285 stars 111 forks source link

Jaxlib #123

Open greeberss opened 2 weeks ago

greeberss commented 2 weeks ago

I haven't changed anything but now I get this error

/storage/stable-diffusion/stable-diffusion-webui Traceback (most recent call last): File "/storage/stable-diffusion/stable-diffusion-webui/webui.py", line 13, in initialize.imports() File "/storage/stable-diffusion/stable-diffusion-webui/modules/initialize.py", line 17, in imports import pytorch_lightning # noqa: F401 File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/init.py", line 35, in from pytorch_lightning.callbacks import Callback # noqa: E402 File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/init.py", line 14, in from pytorch_lightning.callbacks.batch_size_finder import BatchSizeFinder File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/batch_size_finder.py", line 24, in from pytorch_lightning.callbacks.callback import Callback File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/callback.py", line 25, in from pytorch_lightning.utilities.types import STEP_OUTPUT File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/types.py", line 27, in from torchmetrics import Metric File "/usr/local/lib/python3.10/dist-packages/torchmetrics/init.py", line 26, in from torchmetrics import functional # noqa: E402 File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/init.py", line 122, in from torchmetrics.functional.text._deprecated import _bleu_score as bleu_score File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/text/init.py", line 50, in from torchmetrics.functional.text.bert import bert_score File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/text/bert.py", line 40, in from transformers import AutoModel, AutoTokenizer File "/usr/local/lib/python3.10/dist-packages/transformers/init.py", line 26, in from . import dependency_versions_check File "/usr/local/lib/python3.10/dist-packages/transformers/dependency_versions_check.py", line 17, in from .utils.versions import require_version, require_version_core File "/usr/local/lib/python3.10/dist-packages/transformers/utils/init.py", line 30, in from .generic import ( File "/usr/local/lib/python3.10/dist-packages/transformers/utils/generic.py", line 33, in import jax.numpy as jnp File "/usr/local/lib/python3.10/dist-packages/jax/init.py", line 35, in from jax import config as _config_module File "/usr/local/lib/python3.10/dist-packages/jax/config.py", line 17, in from jax._src.config import config # noqa: F401 File "/usr/local/lib/python3.10/dist-packages/jax/_src/config.py", line 25, in from jax._src import lib File "/usr/local/lib/python3.10/dist-packages/jax/_src/lib/init.py", line 85, in cpu_feature_guard.check_cpu_features() RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

Cyberes commented 2 weeks ago 
 RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

Is this Paperspace?

storoonnik commented 2 weeks ago

Just wanted to add that I am also encountering this error, although I've reconfigured my setup to use the Forge fork of WebUI.

/storage/stable-diffusion-webui-forge
Total VRAM 16109 MB, total RAM 45140 MB
xformers version: 0.0.26.post1
Set vram state to: NORMAL_VRAM
Device: cuda:0 NVIDIA RTX A4000 : native
Hint: your device supports --pin-shared-memory for potential speed improvements.
Hint: your device supports --cuda-malloc for potential speed improvements.
Hint: your device supports --cuda-stream for potential speed improvements.
VAE dtype: torch.bfloat16
CUDA Stream Activated:  False
Traceback (most recent call last):
  File "/storage/stable-diffusion-webui-forge/webui.py", line 19, in <module>
    initialize.imports()
  File "/storage/stable-diffusion-webui-forge/modules/initialize.py", line 29, in imports
    import pytorch_lightning  # noqa: F401
  File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/__init__.py", line 30, in <module>
    from pytorch_lightning.callbacks import Callback  # noqa: E402
  File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/__init__.py", line 14, in <module>
    from pytorch_lightning.callbacks.base import Callback
  File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/base.py", line 25, in <module>
    from pytorch_lightning.utilities.types import STEP_OUTPUT
  File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/types.py", line 26, in <module>
    from torchmetrics import Metric
  File "/usr/local/lib/python3.10/dist-packages/torchmetrics/__init__.py", line 26, in <module>
    from torchmetrics import functional  # noqa: E402
  File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/__init__.py", line 122, in <module>
    from torchmetrics.functional.text._deprecated import _bleu_score as bleu_score
  File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/text/__init__.py", line 50, in <module>
    from torchmetrics.functional.text.bert import bert_score
  File "/usr/local/lib/python3.10/dist-packages/torchmetrics/functional/text/bert.py", line 40, in <module>
    from transformers import AutoModel, AutoTokenizer
  File "/usr/local/lib/python3.10/dist-packages/transformers/__init__.py", line 26, in <module>
    from . import dependency_versions_check
  File "/usr/local/lib/python3.10/dist-packages/transformers/dependency_versions_check.py", line 17, in <module>
    from .utils.versions import require_version, require_version_core
  File "/usr/local/lib/python3.10/dist-packages/transformers/utils/__init__.py", line 30, in <module>
    from .generic import (
  File "/usr/local/lib/python3.10/dist-packages/transformers/utils/generic.py", line 33, in <module>
    import jax.numpy as jnp
  File "/usr/local/lib/python3.10/dist-packages/jax/__init__.py", line 35, in <module>
    from jax import config as _config_module
  File "/usr/local/lib/python3.10/dist-packages/jax/config.py", line 17, in <module>
    from jax._src.config import config  # noqa: F401
  File "/usr/local/lib/python3.10/dist-packages/jax/_src/config.py", line 25, in <module>
    from jax._src import lib
  File "/usr/local/lib/python3.10/dist-packages/jax/_src/lib/__init__.py", line 85, in <module>
    cpu_feature_guard.check_cpu_features()
RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

Perhaps Paperspace changed something behind the scenes?

greeberss commented 2 weeks ago 
 RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

Is this Paperspace?

Yes, A4000

Cyberes commented 2 weeks ago

You should contact Paperspace and report your issue. AVX exists in all modern CPUs so something is broken on their end.