search

ostris / ai-toolkit

Various AI scripts. Mostly Stable Diffusion stuff.
MIT License
220 stars 25 forks source link

Slider-training_ModuleNotFoundError+ #21

Open silveroxides opened 6 months ago

silveroxides commented 6 months ago

When running on Colab using your notebook this error is encountered as there is no such module there.



[<ipython-input-7-8519949f2c4c>](https://localhost:8080/#) in <cell line: 1>()
----> 1 run_job(job_to_run)

8 frames

# [/content/ai-toolkit/toolkit/pipelines.py] (https://localhost:8080/#) in <module>
      6 from diffusers import StableDiffusionXLPipeline, StableDiffusionPipeline, LMSDiscreteScheduler
      7 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
----> 8 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_k_diffusion import ModelWrapper
      9 from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput
     10 from diffusers.pipelines.stable_diffusion_xl.pipeline_stable_diffusion_xl import rescale_noise_cfg

ModuleNotFoundError: No module named 'diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_k_diffusion'
Omenizer commented 6 months ago

Same issue, Module not found :(

Thank you for building the colab, would be great if there was a fix!