maxtext.experimental.agent.ckpt_conversion_agent.analysis module

maxtext.experimental.agent.ckpt_conversion_agent.analysis 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.analysis.AnalysisAgent(api_key, dir_path, target_model='gemma3', max_retries=3)[source]#

Bases: BaseAgent

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

analyze_model_structures()[source]#

Analyzes the model structures of MaxText and Hugging Face parameters.