Skip to content

Commit

Permalink
fix image links
Browse files Browse the repository at this point in the history
  • Loading branch information
younesbelkada committed Dec 19, 2022
1 parent fc8e805 commit 96fbf0b
Show file tree
Hide file tree
Showing 7 changed files with 304 additions and 1,243 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ This process is illustrated in the sketch below:


<div style="text-align: center">
<img src="examples/images/trl_overview.png" width="800">
<img src="https://huggingface.co/datasets/trl-internal-testing/example-images/resolve/main/images/trl_overview.png" width="800">
<p style="text-align: center;"> <b>Figure:</b> Sketch of the workflow. </p>
</div>

Expand Down Expand Up @@ -94,7 +94,7 @@ train_stats = ppo_trainer.step([query_tensor[0]], [response_tensor[0]], reward)
For a detailed example check out the notebook `04-gpt2-sentiment-ppo-training.ipynb`, where GPT2 is fine-tuned to generate positive movie reviews. An few examples from the language models before and after optimisation are given below:

<div style="text-align: center">
<img src="nbs/images/table_imdb_preview.png" width="800">
<img src="https://huggingface.co/datasets/trl-internal-testing/example-images/resolve/main/images/table_imdb_preview.png" width="800">
<p style="text-align: center;"> <b>Figure:</b> A few review continuations before and after optimisation. </p>
</div>

Expand Down
230 changes: 0 additions & 230 deletions examples/notebooks/00-core.ipynb

This file was deleted.

Loading

0 comments on commit 96fbf0b

Please sign in to comment.