Add CogView4 model support

[RyanJDick]

Summary

Add support for the CogView4 model in nodes.

Example workflows:

• Text-to-image: cogview4.json
• Inpainting: cogview4_inpaint.json

Example

Expanded prompt:

A whimsical stuffed gnome sits on a golden sandy beach, its plush fabric slightly textured and well-worn. The gnome has a round, cheerful face with a fluffy white beard, a bulbous nose, and a tall, slightly floppy red hat with a few decorative stitching details. It wears a tiny blue vest over a soft, earthy-toned tunic, and its stubby arms grasp a ripe yellow banana with a few brown speckles. The ocean waves gently roll onto the shore in the background, with turquoise water reflecting the warm glow of the late afternoon sun. A few scattered seashells and driftwood pieces are near the gnome, while a colorful beach umbrella and footprints in the sand hint at a lively beach scene. The sky is a soft pastel blend of pink, orange, and light blue, with wispy clouds stretching across the horizon.

Result:

Follow-up work

• Linear UI integration
• CogView4 performs best with expanded prompts. Add an LLM preprocessing node for prompt expansion. See the system prompts here for reference: https://github.com/THUDM/CogView4/blob/962816cc760188032713dc5293c4588d42fe88e5/inference/prompt_optimize.py

Related Issues / Discussions

N/A

QA Instructions

• Regression test FLUX and SD3 inpainting (since the inpainting extension code was consolidated)
• Smoke test text-to-image with progress images for all base model types (since the progress image code was refactored a bit)
• Install CogView4 model via Starter Models tab
• CogView4 appears in the model list
• Test CogView4 text-to-image workflow
• Test CogView4 image-to-image workflow
• Test CogView4 inpainting workflow
• Test that CogView4 denoising can be cancelled mid-execution
• Test CogView4 progress images (they work, but it looks like there's room for improvement with further tuning)
• Test CogView4 partial loading (constrain available VRAM to ~6GB)

Merge Plan

• This PR pins an arbitrary diffusers commit to get access to the CogView4 model. We must decide if we are ok with this, or want to wait for an official release.

Checklist

• The PR has a short but descriptive title, suitable for a changelog
• Tests added / updated (if applicable)
• Documentation added / updated (if applicable)
• Updated What's New copy (if doing a release after this PR)

[psychedelicious]

While reviewing the CogView4 HF repo, I noticed this inference restriction:

Resolution: Width and height must be between 512px and 2048px, divisible by 32, and ensure the maximum number of pixels does not exceed 2^21 px.

See: https://huggingface.co/THUDM/CogView4-6B#inference-requirements-and-model-introduction

This introduces a new type of constraint. You'd expect the max dimensions to be 2048 x 2048, but that is 4,194,304 pixels, which exceeds max pixel count of 2 ^ 21 = 2,097,152. So we may need to make some changes to dimension constraints to support CogView4.

Also, image sizes must be divisible by 32. This needs to be handled in a number of areas.

Note: I'm still downloading the model - slow internet today - so I haven't actually tested yet. Maybe this is a non-issue. Just reviewing the model docs and taking notes.

[psychedelicious]

The max number of pixels requirement seems to be fake news. I can generate largeer images than 1024x2048, though I OOM with 24GB VRAM around 1700x2000 on VAE decode.

I've added checks for the dimensions.

[psychedelicious]

• Tested text to image, image to image, inpainting/outpainting - all working well.
• Added CogView4 Text to Image to default workflows.

This PR has diffusers pinned to a pre-release commit. We are on diffusers==0.31.0 right now - about 5 months old.

Feels risky to merge this and release w/ an unreleased, potentially unstable diffusers dep. Let's wait for the next stable diffusers release and do thorough testing before merging this PR.

Marked as draft to prevent premature merge.

[a-r-r-o-w]

Great to see InvokeAI support for CogView4! We'll try to do a diffusers release asap to unblock this (hopefully next week) 🤗