Traceback (most recent call last):
File "/root/JDiffusion/examples/dreambooth/run_all.py", line 3, in
from JDiffusion.pipelines import StableDiffusionPipeline
File "/root/JDiffusion/python/JDiffusion/init.py", line 2, in
from .models import (
File "/root/JDiffusion/python/JDiffusion/models/init.py", line 1, in
from .autoencoder_kl_jittor import AutoencoderKL
File "/root/JDiffusion/python/JDiffusion/models/autoencoder_kl_jittor.py", line 19, in
from diffusers.loaders import FromOriginalVAEMixin
File "", line 1055, in _handle_fromlist
File "/root/anaconda3/envs/jdiffusion/lib/python3.9/site-packages/diffusers/utils/import_utils.py", line 692, in getattr
module = self._get_module(self._class_to_module[name])
File "/root/anaconda3/envs/jdiffusion/lib/python3.9/site-packages/diffusers/utils/import_utils.py", line 704, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.loaders.single_file because of the following error (look up to see its traceback):
No module named 'torch.cuda.amp'; 'torch.cuda' is not a package
在运行 run_all.py 时遇到问题
Traceback (most recent call last): File "/root/JDiffusion/examples/dreambooth/run_all.py", line 3, in
from JDiffusion.pipelines import StableDiffusionPipeline
File "/root/JDiffusion/python/JDiffusion/init.py", line 2, in
from .models import (
File "/root/JDiffusion/python/JDiffusion/models/init.py", line 1, in
from .autoencoder_kl_jittor import AutoencoderKL
File "/root/JDiffusion/python/JDiffusion/models/autoencoder_kl_jittor.py", line 19, in
from diffusers.loaders import FromOriginalVAEMixin
File "", line 1055, in _handle_fromlist
File "/root/anaconda3/envs/jdiffusion/lib/python3.9/site-packages/diffusers/utils/import_utils.py", line 692, in getattr
module = self._get_module(self._class_to_module[name])
File "/root/anaconda3/envs/jdiffusion/lib/python3.9/site-packages/diffusers/utils/import_utils.py", line 704, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.loaders.single_file because of the following error (look up to see its traceback):
No module named 'torch.cuda.amp'; 'torch.cuda' is not a package
已经确认安装了 torch