Skip to content

Commit b8f089c

Browse files
authored
fix/doc-code: import torch and fix the broken document address (#3941)
Signed-off-by: GitHub <[email protected]>
1 parent 187ea53 commit b8f089c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docs/source/en/using-diffusers/loading.mdx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ A checkpoint variant is usually a checkpoint where it's weights are:
174174

175175
</Tip>
176176

177-
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.
177+
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.
178178

179179
| **checkpoint type** | **weight name** | **argument for loading weights** |
180180
|---------------------|-------------------------------------|----------------------------------|
@@ -190,6 +190,7 @@ There are two important arguments to know for loading variants:
190190

191191
```python
192192
from diffusers import DiffusionPipeline
193+
import torch
193194

194195
# load fp16 variant
195196
stable_diffusion = DiffusionPipeline.from_pretrained(

0 commit comments

Comments
 (0)