Skip to content

GitLab

"...text-generation-inference.git" did not exist on "245d3de94877d4910ea7876f6bd19b4d7d4a47d9"
Unverified Commit 2f0fcf4f authored by Julien Simon's avatar Julien Simon Committed by GitHub
Browse files

Add missing import (#979)

parent cc436087
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ 2f0fcf4f
...@@ -88,6 +88,7 @@ We recommend using the model in [half-precision (`fp16`)](https://pytorch.org/bl ...@@ -88,6 +88,7 @@ We recommend using the model in [half-precision (`fp16`)](https://pytorch.org/bl
precision while being roughly twice as fast and requiring half the amount of GPU RAM. precision while being roughly twice as fast and requiring half the amount of GPU RAM.
```python ```python
import torch
from diffusers import StableDiffusionPipeline from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16") pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment