This repository was archived by the owner on Sep 10, 2025. It is now read-only.
-
Couldn't load subscription status.
- Fork 249
This repository was archived by the owner on Sep 10, 2025. It is now read-only.
Llama3.2 vision model AOTI integration #1497
Open
@larryliu0820
Description
🚀 The feature, motivation and pitch
Need work to enable:
python3 torchchat.py export llama3.2-11B --output-aoti-package-path exportedModels/llama3_2_artifacts.pt2
Features to be added:
- Run AOTI export on preprocess, vision encoder and text decoder separately.
- This requires dependency on torchtune's exportable modules and perform module transformation needed.
- Package them into a single pt2 file.
Alternatives
No response
Additional context
No response
RFC (Optional)
No response
Metadata
Metadata
Assignees
Labels
Type
Projects
Status
Staging