Add Prompt Enhancer and Processor for Standard Pipeline

#15
by dg845 - opened
No description provided.

This PR is a follow up to #13 that adds the prompt_enhancer and processor from google/gemma-4-E2B-it to the repo for the standard pipeline (which unlike the modular pipeline cannot be configured to directly load the weights from the Gemma 4 repo). Unlike #13, it doesn't accidentally duplicate the prompt_enhancer weights by including both model.safetensors and model shards.

You can test the changes by loading the pipeline with the refs/pr/15 revision corresponding to this PR:

import torch
from diffusers import LTX2Pipeline

pipe = LTX2Pipeline.from_pretrained("Lightricks/LTX-2.5-Diffusers", revision="refs/pr/15", torch_dtype=torch.bfloat16)
...
dg845 changed pull request status to open
art-alex changed pull request status to merged

Sign up or log in to comment