Doesn't work in container #359
Description
I created a container and ran it with the following commands:
docker build -t acestep-image .
docker run -p 7865:7865 acestep-image
It doesn't work and it says:
C:\ACE-Step>docker run -p 7865:7865 acestep-image
==========
== CUDA ==
CUDA Version 12.6.0
Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
This container image and its contents are governed by the NVIDIA Deep Learning Container License.
By pulling and using the container, you accept the terms and conditions of this license:
https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license
A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience.
WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available.
Use the NVIDIA Container Toolkit to start this container with GPU support; see
https://docs.nvidia.com/datacenter/cloud-native/ .
Traceback (most recent call last):
File "/opt/venv/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 953, in _get_module
return importlib.import_module("." + module_name, self.name)
File "/usr/lib/python3.10/importlib/init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "", line 1050, in _gcd_import
File "", line 1027, in _find_and_load
File "", line 1006, in _find_and_load_unlocked
File "", line 688, in _load_unlocked
File "", line 883, in exec_module
File "", line 241, in _call_with_frames_removed
File "/opt/venv/lib/python3.10/site-packages/diffusers/loaders/lora_pipeline.py", line 34, in
from .lora_base import ( # noqa
File "/opt/venv/lib/python3.10/site-packages/diffusers/loaders/lora_base.py", line 56, in
from peft.tuners.tuners_utils import BaseTunerLayer
File "/opt/venv/lib/python3.10/site-packages/peft/init.py", line 17, in
from .auto import (
File "/opt/venv/lib/python3.10/site-packages/peft/auto.py", line 32, in
from .peft_model import (
File "/opt/venv/lib/python3.10/site-packages/peft/peft_model.py", line 42, in
from peft.tuners.lora.variants import get_alora_offsets_for_forward, get_alora_offsets_for_generate
File "/opt/venv/lib/python3.10/site-packages/peft/tuners/init.py", line 15, in
from .adalora import AdaLoraConfig, AdaLoraModel
File "/opt/venv/lib/python3.10/site-packages/peft/tuners/adalora/init.py", line 18, in
from .config import AdaLoraConfig
File "/opt/venv/lib/python3.10/site-packages/peft/tuners/adalora/config.py", line 19, in
from peft.tuners.lora import LoraConfig
File "/opt/venv/lib/python3.10/site-packages/peft/tuners/lora/init.py", line 23, in
from .model import LoraModel
File "/opt/venv/lib/python3.10/site-packages/peft/tuners/lora/model.py", line 26, in
from transformers.modeling_layers import GradientCheckpointingLayer
ModuleNotFoundError: No module named 'transformers.modeling_layers'
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "/app/acestep/gui.py", line 79, in
main()
File "/opt/venv/lib/python3.10/site-packages/click/core.py", line 1485, in call
return self.main(*args, **kwargs)
File "/opt/venv/lib/python3.10/site-packages/click/core.py", line 1406, in main
rv = self.invoke(ctx)
File "/opt/venv/lib/python3.10/site-packages/click/core.py", line 1269, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/opt/venv/lib/python3.10/site-packages/click/core.py", line 824, in invoke
return callback(*args, **kwargs)
File "/app/acestep/gui.py", line 58, in main
from acestep.pipeline_ace_step import ACEStepPipeline
File "/opt/venv/lib/python3.10/site-packages/acestep/pipeline_ace_step.py", line 31, in
from diffusers.pipelines.stable_diffusion_3.pipeline_stable_diffusion_3 import (
File "/opt/venv/lib/python3.10/site-packages/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py", line 30, in
from ...loaders import FromSingleFileMixin, SD3IPAdapterMixin, SD3LoraLoaderMixin
File "", line 1075, in _handle_fromlist
File "/opt/venv/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 943, in getattr
module = self._get_module(self._class_to_module[name])
File "/opt/venv/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 955, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.loaders.lora_pipeline because of the following error (look up to see its traceback):
No module named 'transformers.modeling_layers'
It should have packed all the dependencies, right? It should have worked, right?