在Google Colab运行DreamBooth时遭遇ModuleNotFoundError求助

我使用TheLastBen的fast-DreamBooth Colab笔记本，通过Automatic1111 WebUI运行DreamBooth。之前一切正常，但最近训练模型后启动WebUI时，出现找不到torchsde模块的错误。我试过替换sampling.py等文件但没解决；删除所有文件重新训练后，又出现找不到safetensors模块的错误。因为本地GPU是NVIDIA 940 MX（2GB）性能有限，只能依赖Colab生成自定义图像训练的DreamBooth作品，求解决办法。

第一次错误回溯 #

Traceback (most recent call last):
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/webui.py", line 13, in <module>
    from modules import shared, devices, sd_samplers, upscaler, extensions, localization, ui_tempdir
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_samplers.py", line 8, in <module>
    import k_diffusion.sampling
  File "/content/gdrive/MyDrive/sd/stable-diffusion/src/k-diffusion/k_diffusion/__init__.py", line 1, in <module>
    from . import augmentation, config, evaluation, external, gns, layers, models, sampling, utils
  File "/content/gdrive/MyDrive/sd/stable-diffusion/src/k-diffusion/k_diffusion/external.py", line 6, in <module>
    from . import sampling, utils
  File "/content/gdrive/MyDrive/sd/stable-diffusion/src/k-diffusion/k_diffusion/sampling.py", line 7, in <module>
    import torchsde
ModuleNotFoundError: No module named 'torchsde'

重新训练后的新错误回溯 #

Traceback (most recent call last):
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/webui.py", line 13, in <module>
    from modules import shared, devices, sd_samplers, upscaler, extensions, localization, ui_tempdir
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_samplers.py", line 11, in <module>
    from modules import prompt_parser, devices, processing, images
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/processing.py", line 15, in <module>
    import modules.sd_hijack
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_hijack.py", line 10, in <module>
    import modules.textual_inversion.textual_inversion
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/textual_inversion/textual_inversion.py", line 13, in <module>
    from modules import shared, devices, sd_hijack, processing, sd_models, images, sd_samplers
  File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_models.py", line 8, in <module>
    import safetensors.torch
ModuleNotFoundError: No module named 'safetensors'

• 直接在Colab的代码单元格中安装缺失的依赖：
  • 安装torchsde：

    !pip install torchsde
    

  • 安装safetensors：

    !pip install safetensors
    

• 如果安装后仍报错，检查WebUI环境与训练环境是否一致，启动WebUI前重新安装所有依赖：

  !cd /content/gdrive/MyDrive/sd/stable-diffusion-webui && pip install -r requirements.txt
  

• 不要手动替换sampling.py这类核心文件，避免版本不兼容问题。建议使用Colab笔记本自带的更新/重置功能，恢复官方的k-diffusion和WebUI文件。

内容的提问来源于stack exchange，提问作者Huzaifa Arshad