세팅 다하고 최종 ./my_training.sh 호출하는데 에러뜸... 일단 my_training.sh 내용은 이거고
export MODEL_NAME="latest"
export INSTANCE_DIR="training/test"
export CLASS_DIR="classes"
export OUTPUT_DIR="complte"
accelerate launch train_dreambooth.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--instance_data_dir=$INSTANCE_DIR \
--class_data_dir=$CLASS_DIR \
--output_dir=$OUTPUT_DIR \
--instance_prompt="test" \
--class_prompt="test" \
--resolution=512 \
--train_batch_size=1 \
--mixed_precision="fp16" \
--use_8bit_adam \
--gradient_accumulation_steps=1 --gradient_checkpointing \
--learning_rate=5e-6 \
--lr_scheduler="constant" \
--lr_warmup_steps=0 \
--max_train_steps=2500
에러는 아래인데
Traceback (most recent call last):
File "/home/flux/github/diffusers/examples/dreambooth/train_dreambooth.py", line 765, in <module>
main()
File "/home/flux/github/diffusers/examples/dreambooth/train_dreambooth.py", line 489, in main
import bitsandbytes as bnb
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/__init__.py", line 6, in <module>
from .autograd._functions import (
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/autograd/_functions.py", line 5, in <module>
import bitsandbytes.functional as F
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/functional.py", line 13, in <module>
from .cextension import COMPILED_WITH_CUDA, lib
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/cextension.py", line 113, in <module>
lib = CUDASetup.get_instance().lib
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/cextension.py", line 109, in get_instance
cls._instance.initialize()
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/cextension.py", line 59, in initialize
binary_name, cudart_path, cuda, cc, cuda_version_string = evaluate_cuda_setup()
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/cuda_setup/main.py", line 125, in evaluate_cuda_setup
cudart_path = determine_cuda_runtime_lib_path()
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/bitsandbytes/cuda_setup/paths.py", line 90, in determine_cuda_runtime_lib_path
CUDASetup.get_instance.add_log_entry(f'{candidate_env_vars["CONDA_PREFIX"]} did not contain '
AttributeError: 'function' object has no attribute 'add_log_entry'
Traceback (most recent call last):
File "/home/flux/anaconda3/envs/diffusers/bin/accelerate", line 8, in <module>
sys.exit(main())
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/accelerate/commands/accelerate_cli.py", line 43, in main
args.func(args)
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/accelerate/commands/launch.py", line 837, in launch_command
simple_launcher(args)
File "/home/flux/anaconda3/envs/diffusers/lib/python3.9/site-packages/accelerate/commands/launch.py", line 354, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/home/flux/anaconda3/envs/diffusers/bin/python'
ㄹㅇ 개답답하네 디버깅할려고해도 파이썬은 잘모르겟다