Skip to content

Commit 6adfedf

Browse files
Apply suggestions from code review
1 parent 94781b6 commit 6adfedf

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

docs/source/en/optimization/fp16.mdx

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,9 +162,7 @@ images = pipe([prompt], width=3840, height=2224).images
162162
The output image will have some tile-to-tile tone variation from the tiles having separate decoders, but you shouldn't see sharp seams between the tiles. The tiling is turned off for images that are 512x512 or smaller.
163163

164164

165-
=======
166165
<a name="sequential_offloading"></a>
167-
>>>>>>> 664b4de9e22a825875b6bad45867f8f77cdf95d6:docs/source/en/optimization/fp16.mdx
168166
## Offloading to CPU with accelerate for memory savings
169167

170168
For additional memory savings, you can offload the weights to CPU and only load them to GPU when performing the forward pass.

tests/pipelines/stable_diffusion/test_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -422,7 +422,7 @@ def test_stable_diffusion_vae_slicing(self):
422422
def test_stable_diffusion_vae_tiling(self):
423423
device = "cpu" # ensure determinism for the device-dependent torch.Generator
424424
unet = self.dummy_cond_unet
425-
scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear")
425+
scheduler = LMSDiscreteScheduler.from_pretrained("CompVis/stable-diffusion-v1-4")
426426
vae = self.dummy_vae
427427
bert = self.dummy_text_encoder
428428
tokenizer = CLIPTokenizer.from_pretrained("hf-internal-testing/tiny-random-clip")

0 commit comments

Comments
 (0)