import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_id = "nabin2004/qwen-Manimator-1-merged"tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto") messages = [ {"role": "system", "content": "You are an expert Python programmer and mathematics educator specializing in ManimCE and Manim Voiceover..."}, {"role": "user", "content": "Create a Manim animation explaining Fourier series with audio narration."},]ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to("cuda")out = model.generate(ids, max_new_tokens=2048, temperature=0.2, top_p=0.9)print(tokenizer.decode(out[0][ids.shape[-1]:], skip_special_tokens=True))