Disty0
/

sote-diffusion-cascade_pre-alpha0

StableCascadePriorPipeline

Model card Files Files and versions

Disty0 commited on Mar 17, 2024

Commit

bcb10e8

·

verified ·

1 Parent(s): 3e4b88f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -47,7 +47,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()
@@ -55,7 +55,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]