我正在使用 python 3.11 和 cuda 12.1 ..我正在尝试在本地计算机上训练模型。但安装后出现错误..
!pip install -U autotrain-advanced
!autotrain setup --update-torch 我已经完成了上述命令,但是当我给出时 autotrain dreambooth --model stableai/stable-diffusion-xl-base-1.0 --image-path ayaz/ --prompt "photo of ayaz1 person" --resolution 1024 --batch-size 1 --num-steps 100 --梯度累积 4 --lr 1e-4 --项目名称 my-autotrain-project 它显示错误
autotrain dreambooth --model stableai/stable-diffusion-xl-base-1.0 --image-path ayaz/ --prompt "photo of ayaz1 person" --resolution 1024 --batch-size 1 -- num-steps 100 --gradient-accumulation 4 --lr 1e-4 --project-name my-autotrain-project 回溯(最近一次调用最后一次): 文件“”,第 198 行,位于 _run_module_as_main 中 文件“”,第 88 行, run_code 文件“C:\Ayaz\My_Projects\Prj_general\venv_general\Scripts\autotrain.exe_ main .py”,第 7 行,位于 文件“C:\Ayaz\My_Projects\Prj_general\venv_general\Lib\site-packages\autotrain\cli\autotrain.py”,第 67 行,在 main 中 命令 = args.func(args) ^^^^^^^^^^^^^^^^ 文件“C:\Ayaz\My_Projects\Prj_general\venv_general\Lib\site-packages\autotrain\cli\run_dreambooth.py”,第 21 行,在 run_dreambooth_command_factory 中 返回 RunAutoTrainDreamboothCommand(args) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件“C:\Ayaz\My_Projects\Prj_general\venv_general\Lib\site-packages\autotrain\cli\run_dreambooth.py”,第 365 行, init raise ValueError(“❌ 请指定有效的图像目录” ”) ValueError: ❌ 请指定有效的图像目录
(venv_general) C:\Ayaz\My_Projects\Prj_general\Test> (venv_general) C:\Ayaz\My_Projects\Prj_general\Test>cd " C:\Ayaz\My_Projects\Prj_general" 文件名、目录名或卷标语法不正确。
(venv_general) C:\Ayaz\My_Projects\Prj_general\Test> cd "C:\Ayaz\My_Projects\Prj_general"
(venv_general) C:\ Ayaz\My_Projects\Prj_general>autotrain dreambooth --model stableai/stable-diffusion-xl-base-1.0 --image-path ayaz/ --prompt "photo of ayaz1 person" --resolution 1024 --batch-size 1 -- num-steps 100 --gradient-accumulation 4 --lr 1e-4 --project-name my-autotrain-project 信息| 2024-08-09 10:20:43 | autotrain.cli.run_dreambooth:run:388 - 运行 DreamBooth 训练 警告| 2024-08-09 10:20:43 | autotrain.trainers.common: init :180 - 提供但未使用的参数:推理、日志、版本、部署、train_split、valid_split、func、train、data_path、config、backend 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20160426_182736.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20160912_071125.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20160920_133005(0).jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20160920_133014.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20161017_110932.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20161017_113647.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20161017_141325.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:68 - ayaz/20161019_102543.jpg 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth:_save_concept_images:67 - 保存概念图像 信息| 2024-08-09 10:20:43 | autotrain.preprocessor.dreambooth: save_concept_images:68 - ayaz/20161019_102546.jpg 信息| 2024-08-09 10:20:43 | autotrain.backends.local:create:8 - 开始本地训练... 信息| 2024-08-09 10:20:43 | autotrain.commands:launch_command:478 - ['python', '-m', 'autotrain.trainers.dreambooth', '--training_config', 'my-autotrain-project\training_params.json'] 信息| 2024-08-09 10:20:43 | autotrain.commands:launch_command:479 - {'model':'stabilityai/stable-diffusion-xl-base-1.0','vae_model':无,'revision':无,'tokenizer':无 ne,'image_path':'my-autotrain-project / autotrain-data','class_image_path':无,'prompt':'ayaz1 人的照片','class_prompt':无,'num_class_images':100,'class_labels | ||调节':无,'prior_preservation':False,'prior_loss_weight':1.0,'project_name':'my-autotrain-project','种子':42,'分辨率':1024,'center_crop':False,'火车_te xt_encoder':假,'batch_size':1,'sample_batch_size':4,'epochs':1,'num_steps':100,'checkpointing_steps':100000,'resume_from_checkpoint':无,'gradient_accumulation': 4、'disable_gradient_checkpointing': False、'lr': 0.0001、'scale_lr': False、'scheduler': 'constant'、'warmup_steps': 0、'num_cycles': 1、'lr_power': 1.0、'dataloader_num_workers ': 0, 'use_8bit_adam': False, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'adam_weight_decay': 0.01, 'adam_epsilon': 1e-08, 'max_grad_norm': 1.0, 'allow_tf32': False, 'prior_generatio n_ precision':无,'local_rank':-1,'xformers':False,'pre_compute_text_embeddings':False,'tokenizer_max_length':无,'text_encoder_use_attention_mask':False,'rank':4,' xl': True, 'mixed_ precision': 'none', 'token': None, 'push_to_hub': False, 'username': None, 'validation_prompt': None, 'num_validation_images': 4, 'validation_epochs': 50, ' checkpoints_total_limit':无,'validation_images':无,'logging':False} C:\Users\Gaming PC\AppData\Local\Programs\Python\Python312\python.exe:查找“autotrain.trainers.dreambooth”的模块规范时出错(ModuleNotFoundError:没有名为“autotrain”的模块) 我有尝试了不同的Python版本。
i have tried on different python version.
似乎在虚拟环境中运行autotrain时遇到了一些问题。错误信息表明Python找不到“autotrain”模块。以下是一些可能的解决方案:
1. 检查虚拟环境
-
确认已激活虚拟环境。
可以在终端提示符中看到环境名称。
bash (venv_general) C:\Ayaz\My_Projects\Prj_general>
-
如果未激活,请使用以下命令激活:
bash venv_general\Scripts\activate
-
确认已在虚拟环境中安装了autotrain。
bash pip show autotrain-advanced
如果未安装,请使用以下命令安装:bash pip install autotrain-advanced
2. 检查图像路径
错误信息还指出“❌ 请指定有效的图像目录”。确保
--image-path
参数指向包含训练图像的目录。
*
使用正确的路径:
使用
ayaz/
还是
.\ayaz\
取决于运行脚本的目录和图像目录的位置。如果“ayaz”文件夹与的脚本位于同一目录,请使用
.\ayaz\
。
*
使用绝对路径:
为避免混淆,请尝试使用图像文件夹的完整路径,例如
C:\Ayaz\My_Projects\Prj_general\ayaz\
。
3. 创建新虚拟环境(可选)
如果上述步骤不起作用,请尝试创建一个新的虚拟环境并重新安装所有软件包:
1.
创建新环境:
bash
python -m venv venv_general
2.
激活环境:
bash
venv_general\Scripts\activate
3.
安装autotrain:
bash
pip install autotrain-advanced
4.
再次运行的命令。
4. 其他提示
- 确保的CUDA版本与autotrain兼容。 可以在autotrain文档中找到支持的CUDA版本列表。
-
更新pip:
bash python -m pip install --upgrade pip
- 尝试使用不同的Python版本。 尽管提到尝试了不同的版本,但请确保使用的是支持的版本,并且已在该版本的环境中安装了autotrain。
如果问题仍然存在,请提供以下信息,以便我为提供更具体的帮助: * 使用的操作系统 * 使用的autotrain版本 * 完整的错误消息 * 的训练脚本
希望这些建议能帮助解决问题!
标签:python,stable-diffusion From: 78851453