Skip to content

Commit 6bd80ab

Browse files
committed
[Tutorial] LLM integration
ghstack-source-id: c354a96 Pull Request resolved: #2832
1 parent 6ea8419 commit 6bd80ab

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

torchrl/envs/utils.py

-1
Original file line numberDiff line numberDiff line change
@@ -944,7 +944,6 @@ def make_shape(shape):
944944
else NonTensor(
945945
shape=tensor.shape,
946946
example_data=tensor.data,
947-
example_data=data.data,
948947
device=tensor.device,
949948
)
950949
if is_non_tensor(tensor)

tutorials/sphinx-tutorials/llm.py

+17
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
"""
2+
Interacting with LLMs in TorchRL
3+
================================
4+
5+
**Author**: `Vincent Moens <https://github.com/vmoens>`_
6+
7+
.. _gs_first_training:
8+
9+
.. note:: To run this tutorial in a notebook, add an installation cell
10+
at the beginning containing:
11+
12+
.. code-block::
13+
14+
!pip install tensordict
15+
!pip install torchrl
16+
17+
"""

0 commit comments

Comments
 (0)