Unverified Commit b8f089c5 authored by Aisuko's avatar Aisuko Committed by GitHub
Browse files

fix/doc-code: import torch and fix the broken document address (#3941)


Signed-off-by: default avatarGitHub <noreply@github.com>
parent 187ea539
...@@ -174,7 +174,7 @@ A checkpoint variant is usually a checkpoint where it's weights are: ...@@ -174,7 +174,7 @@ A checkpoint variant is usually a checkpoint where it's weights are:
</Tip> </Tip>
Otherwise, a variant is **identical** to the original checkpoint. They have exactly the same serialization format (like [Safetensors](./using-diffusers/using_safetensors)), model structure, and weights have identical tensor shapes. Otherwise, a variant is **identical** to the original checkpoint. They have exactly the same serialization format (like [Safetensors](./using_safetensors)), model structure, and weights have identical tensor shapes.
| **checkpoint type** | **weight name** | **argument for loading weights** | | **checkpoint type** | **weight name** | **argument for loading weights** |
|---------------------|-------------------------------------|----------------------------------| |---------------------|-------------------------------------|----------------------------------|
...@@ -190,6 +190,7 @@ There are two important arguments to know for loading variants: ...@@ -190,6 +190,7 @@ There are two important arguments to know for loading variants:
```python ```python
from diffusers import DiffusionPipeline from diffusers import DiffusionPipeline
import torch
# load fp16 variant # load fp16 variant
stable_diffusion = DiffusionPipeline.from_pretrained( stable_diffusion = DiffusionPipeline.from_pretrained(
......
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