INNER CODE UNIT · Python

process_text_to_lottie

OpenVGLab/OmniLottie · app.py:583

def process_text_to_lottie(text_prompt, max_tokens, use_sampling, temperature, top_p, top_k):
    with generation_lock:
        try:
            start_time = time.time()

            if not text_prompt or not text_prompt.strip():
                return None, "❌ Please enter a text description", None
            model, processor, device = load_model_once()

            messages = build_messages("text", text_prompt=text_prompt)
            inputs = prepare_inference_input(processor, messages, device)

            generated_ids = generate_lottie(
                model, inputs, max_tokens, device,
                use_sampling, temperature, top_p, top_k
            )

            del inputs

View source record →