From 66316024c7ce468e703c34703ae8833ec1016405 Mon Sep 17 00:00:00 2001 From: Aisuko Date: Tue, 4 Jul 2023 10:08:39 +0000 Subject: [PATCH] fix/doc-code: import torch and fix the broken document address Signed-off-by: GitHub --- docs/source/en/using-diffusers/loading.mdx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/source/en/using-diffusers/loading.mdx b/docs/source/en/using-diffusers/loading.mdx index 8ebd3569e4b0..79c8b278468d 100644 --- a/docs/source/en/using-diffusers/loading.mdx +++ b/docs/source/en/using-diffusers/loading.mdx @@ -174,7 +174,7 @@ A checkpoint variant is usually a checkpoint where it's weights are: -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** | |---------------------|-------------------------------------|----------------------------------| @@ -190,6 +190,7 @@ There are two important arguments to know for loading variants: ```python from diffusers import DiffusionPipeline +import torch # load fp16 variant stable_diffusion = DiffusionPipeline.from_pretrained(