maxtext.experimental.agent.ckpt_conversion_agent.plan module

A plan agent to analysis the HF & Maxtext models architecture and generate a conversion plan in json format.

class maxtext.experimental.agent.ckpt_conversion_agent.plan.PlanAgent(api_key, dir_path, target_model='gemma3', max_retries=3)

Bases: BaseAgent

An agent that demonstrates a multi-step prompt chain to generate a model conversion script, with verification that every parameter is mapped.

plan_conversion()

Json Plan