smthem/

ltx-2-19b-dev-diffusers-4bit

from diffusers import LTX2Pipeline
from diffusers.pipelines.ltx2.export_utils import encode_video
repo= 'smthem/ltx-2-19b-dev-diffusers-4bit'


### text_encoder
from transformers import Gemma3ForConditionalGeneration
text_encoder = Gemma3ForConditionalGeneration.from_pretrained(
                  repo,
                  subfolder="text_encoder",
                )

### transformer
transformer_4bit = AutoModel.from_pretrained(
                    repo,
                    subfolder="transformer",
                )
pipeline = LTX2Pipeline.from_pretrained("smthem/ltx-2-19b-dev-diffusers-test",transformer=transformer_4bit,text_encoder=text_encoder,torch_dtype=torch.float16,)
pipeline.enable_model_cpu_offload()

prompt='A video of a dog dancing to energetic electronic dance music'
negative_prompt="blurry, out of focus, overexposed, underexposed, low contrast, washed out colors, excessive noise, "
            "grainy texture, poor lighting, flickering, motion blur, distorted proportions, unnatural skin tones, "
            "deformed facial features, asymmetrical face, missing facial features, extra limbs, disfigured hands, "
            "wrong hand count, artifacts around text, inconsistent perspective, camera shake, incorrect depth of "
            "field, background too sharp, background clutter, distracting reflections, harsh shadows, inconsistent "
            "lighting direction, color banding, cartoonish rendering, 3D CGI look, unrealistic materials, uncanny "
            "valley effect, incorrect ethnicity, wrong gender, exaggerated expressions, wrong gaze direction, "
            "mismatched lip sync, silent or muted audio, distorted voice, robotic voice, echo, background noise, "
            "off-sync audio,incorrect dialogue, added dialogue, repetitive speech, jittery movement, awkward "
            "pauses, incorrect timing, unnatural transitions, inconsistent framing, tilted camera, flat lighting, "
            "inconsistent tone, cinematic oversaturation, stylized filters, or AI artifacts."


video, audio = pipeline(
      prompt=prompt,
      negative_prompt=negative_prompt,
      height=512,
      width=768,
      num_frames=121,
      frame_rate=25,
      num_inference_steps=20,
      guidance_scale=guidance_scale,
      generator=torch.Generator(device="cuda").manual_seed(42),
      output_type="np",
      return_dict=False,
  )

# Convert video to uint8 (but keep as NumPy array)
video = (video * 255).round().astype("uint8")
video = torch.from_numpy(video)

encode_video(
      video[0],
      fps=args.frame_rate,
      audio=audio[0].float().cpu(),
      audio_sample_rate=pipeline.vocoder.config.output_sampling_rate,  # should be 24000
      output_path=os.path.join(args.output_dir, args.output_filename),
  )