stabilityai
/

stable-cascade

StableCascadeDecoderPipeline

Model card Files Files and versions

dome272 commited on Feb 13, 2024

Commit

4cc5667

·

verified ·

1 Parent(s): 90a3d8f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -85,7 +85,7 @@ prompt = "Anthropomorphic cat dressed as a pilot"
 negative_prompt = ""
 with torch.cuda.amp.autocast(dtype=dtype):
-    prior_output = prior_pipeline(
         prompt=prompt,
         height=1024,
         width=1024,
@@ -93,7 +93,7 @@ with torch.cuda.amp.autocast(dtype=dtype):
         guidance_scale=4.0,
         num_images_per_prompt=num_images_per_prompt,
     )
-    decoder_output = decoder_pipeline(
         image_embeddings=prior_output.image_embeddings,
         prompt=prompt,
         negative_prompt=negative_prompt,

 negative_prompt = ""
 with torch.cuda.amp.autocast(dtype=dtype):
+    prior_output = prior(
         prompt=prompt,
         height=1024,
         width=1024,
         guidance_scale=4.0,
         num_images_per_prompt=num_images_per_prompt,
     )
+    decoder_output = decoder(
         image_embeddings=prior_output.image_embeddings,
         prompt=prompt,
         negative_prompt=negative_prompt,