Skip to content

Commit 3eaab67

Browse files
authored
Merge branch 'main' into jafraust/ddp
2 parents 932f3b7 + c4d9d93 commit 3eaab67

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.ci/docker/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ pandocfilters==1.5.1
1616
markdown==3.8.2
1717

1818
# PyTorch Theme
19-
-e git+https://github.com/pytorch/pytorch_sphinx_theme.git@67aa61448e33cae9a345e012be355f8030d3c4f5#egg=pytorch_sphinx_theme2
19+
-e git+https://github.com/pytorch/pytorch_sphinx_theme.git@3066b6f62b0da01c52570c928281bca68287933d#egg=pytorch_sphinx_theme2
2020

2121
# Tutorial dependencies
2222
tqdm==4.66.1

recipes_source/regional_aot.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
Reducing AoT cold start compilation time with regional compilation
44
============================================================================
55
6-
**Author:** `Sayak Paul <https://github.com/sayakpaul>`_, `Charles Bensimon <https://github.com/cbensimon>`_, `Angela Yi <https://github.com/angelayi>`_
6+
**Author:** `Sayak Paul <https://huggingface.co/sayakpaul>`_, `Charles Bensimon <https://huggingface.co/cbensimon>`_, `Angela Yi <https://github.com/angelayi>`_
77
88
In the `regional compilation recipe <https://docs.pytorch.org/tutorials/recipes/regional_compilation.html>`__, we showed
99
how to reduce cold start compilation times while retaining (almost) full compilation benefits. This was demonstrated for
@@ -236,4 +236,6 @@ def aot_compile_load_model(regional=False) -> torch.nn.Module:
236236
#
237237
# This recipe shows how to control the cold start time when compiling your
238238
# model ahead-of-time. This becomes effective when your model has repeated
239-
# blocks, which is typically seen in large generative models.
239+
# blocks, which is typically seen in large generative models. We used this
240+
# recipe on various models to speed up real-time performance. Learn more
241+
# `here <https://huggingface.co/blog/zerogpu-aoti>`__.

0 commit comments

Comments
 (0)