Skip to content

Commit

Permalink
[docs] remove duplicate tip block. (#7625)
Browse files Browse the repository at this point in the history
remove duplicate tip block.
  • Loading branch information
sayakpaul authored Apr 10, 2024
1 parent b99b161 commit a402431
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions docs/source/en/training/distributed_inference.md
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ For example, if you have two 8GB GPUs, then using [`~DiffusionPipeline.enable_mo

To make use of both GPUs, you can use the "balanced" device placement strategy which splits the models across all available GPUs.

> [!TIP]
> [!WARNING]
> Only the "balanced" strategy is supported at the moment, and we plan to support additional mapping strategies in the future.
```diff
Expand All @@ -81,9 +81,6 @@ image = pipeline("a dog").images[0]
image
```

> [!WARNING]
> Currently, we support only "balanced" `device_map`. We plan to support more device mapping strategies in future.
You can also pass a dictionary to enforce the maximum GPU memory that can be used on each device:

```diff
Expand Down

0 comments on commit a402431

Please sign in to comment.