search

Koishi-Star / Euler-Smea-Dyn-Sampler

A sampler base on Euler, aim at generating better picture/一种基于Euler的采样方法，旨在生成更好的图片
Apache License 2.0
158 stars 13 forks source link

Universal extension for WebUI/ComfyUI + Fix inpainting #10

Closed pamparamm closed 5 months ago

pamparamm commented 5 months ago

This should fix issues with tensor shape errors during inpainting. Contains nearest-exact change from #5, can remove it if necessary.

Refactors code quite a lot due to differences in inpainting process in various backends.

  1. Everything from sampling.py is moved into scripts/smea_sampling.py, duplicate code from __init__.py is removed
  2. __init__.py contains sampler registration logic for ComfyUI
  3. scripts/smea.py contains sampler registration logic for WebUI/Forge
  4. scripts/smea_utils.py contains some static variables along with rescaler context manager for masks/noise

Don't know if mask/noise interpolation is working correctly, but at least it doesn't throw any exceptions.

Tested on ComfyUI and on Forge backends

Koishi-Star commented 5 months ago

Thank you for your work! I love it.