-
-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Open
Description
We have two branches of new, especially precise and “clean” VAEs for SDXL called EQ VAE (one from https://huggingface.co/KBlueLeaf/EQ-SDXL-VAE ). Test trainings show a visually noticeable improvement when using these new VAEs for training.
Problems:
- Anzhc’s EQ VAE (https://huggingface.co/Anzhc/MS-LC-EQ-D-VR_VAE) loads correctly, but also requires the patch from the project page to dodge the padding bug:
for module in self.model.modules():
if isinstance(module, nn.Conv2d):
pad_h, pad_w = module.padding if isinstance(module.padding, tuple) else (module.padding, module.padding)
if pad_h > 0 or pad_w > 0:
module.padding_mode = "reflect"
Additional module for the inference fix via Comfy with models trained on Anzhc’s EQ VAE here https://github.com/Jelosus2/comfyui-vae-reflection
It would be great to be able to use in the full extent, if that’s not too much trouble to implement.
muooon
Metadata
Metadata
Assignees
Labels
No labels