Disty0
/

sote-diffusion-cascade-decoder_pre-alpha0

StableCascadeDecoderPipeline

Model card Files Files and versions

Disty0 commited on Mar 17, 2024

Commit

3724f64

·

verified ·

1 Parent(s): ba19c8c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ prior_output = prior(
 decoder.enable_model_cpu_offload()
 decoder_output = decoder(
-    image_embeddings=prior_output.image_embeddings.to(torch.float16),
     prompt=prompt,
     negative_prompt=negative_prompt,
     guidance_scale=2.0,

 decoder.enable_model_cpu_offload()
 decoder_output = decoder(
+    image_embeddings=prior_output.image_embeddings,
     prompt=prompt,
     negative_prompt=negative_prompt,
     guidance_scale=2.0,