Skip to content

Commit d8ec9ae

Browse files
committed
Update base for Update on "[llama-mm] Onboard torchtune vision encoder to ExecuTorch/AOTI"
Summary: As titled. This PR adds `llama3_2_vision_encoder` to `examples/models/llama3_2_vision/vision_encoder` and add CI jobs. Test Plan: Rely on newly added CI jobs Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
1 parent b8b3299 commit d8ec9ae

File tree

3 files changed

+10
-6
lines changed

3 files changed

+10
-6
lines changed

.ci/docker/ci_commit_pins/pytorch.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
19eff28ff3f19b50da46f5a9ff5f4d4d213806fe
1+
c8a648d4dffb9f0133ff4a2ea0e660b42105d3ad

extension/llm/modules/test/test_position_embeddings.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -164,17 +164,21 @@ def test_tiled_token_positional_embedding_export(self):
164164
assert_close(y, ref_y)
165165

166166
def test_tiled_token_positional_embedding_aoti(self):
167-
so = torch._export.aot_compile(
167+
tpe_ep = torch.export.export(
168168
self.tpe,
169-
args=(self.x, self.aspect_ratio),
170-
options={"aot_inductor.package": True},
169+
(self.x, self.aspect_ratio),
171170
dynamic_shapes=(
172171
self.dynamic_shape,
173172
None,
174173
), # assuming aspect ratio is static
175174
)
175+
176176
with tempfile.TemporaryDirectory() as tmpdir:
177-
path = package_aoti(os.path.join(tmpdir, "tpe.pt2"), so)
177+
path = torch._inductor.aoti_compile_and_package(
178+
tpe_ep,
179+
(self.x, self.aspect_ratio),
180+
package_path=os.path.join(tmpdir, "tpe.pt2"),
181+
)
178182
tpe_aoti = load_package(path)
179183

180184
y = tpe_aoti(self.x, self.aspect_ratio)

install_requirements.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def python_is_compatible():
112112
# NOTE: If a newly-fetched version of the executorch repo changes the value of
113113
# NIGHTLY_VERSION, you should re-run this script to install the necessary
114114
# package versions.
115-
NIGHTLY_VERSION = "dev20241112"
115+
NIGHTLY_VERSION = "dev20241101"
116116

117117
# The pip repository that hosts nightly torch packages.
118118
TORCH_NIGHTLY_URL = "https://download.pytorch.org/whl/nightly/cpu"

0 commit comments

Comments
 (0)