Disty0
/

sote-diffusion-cascade-decoder_pre-alpha0

StableCascadeDecoderPipeline

Model card Files Files and versions

Disty0 commited on Mar 17, 2024

Commit

ba19c8c

·

verified ·

1 Parent(s): 9b0caca

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,7 +34,7 @@ prior_output = prior(
     negative_prompt=negative_prompt,
     guidance_scale=6.0,
     num_images_per_prompt=1,
-    num_inference_steps=30
 )
 decoder.enable_model_cpu_offload()
@@ -42,7 +42,7 @@ decoder_output = decoder(
     image_embeddings=prior_output.image_embeddings.to(torch.float16),
     prompt=prompt,
     negative_prompt=negative_prompt,
-    guidance_scale=1.0,
     output_type="pil",
     num_inference_steps=10
 ).images[0]

     negative_prompt=negative_prompt,
     guidance_scale=6.0,
     num_images_per_prompt=1,
+    num_inference_steps=40
 )
 decoder.enable_model_cpu_offload()
     image_embeddings=prior_output.image_embeddings.to(torch.float16),
     prompt=prompt,
     negative_prompt=negative_prompt,
+    guidance_scale=2.0,
     output_type="pil",
     num_inference_steps=10
 ).images[0]