stabilityai
/

stable-cascade

StableCascadeDecoderPipeline

Model card Files Files and versions

multimodalart HF Staff commited on Feb 12, 2024

Commit

d6b6de9

·

verified ·

1 Parent(s): 0a8a23c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -78,7 +78,7 @@ device = "cuda"
 dtype = torch.bfloat16
 num_images_per_prompt = 2
-prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=dtype).to(device)
 decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype).to(device)
 prompt = "Anthropomorphic cat dressed as a pilot"

 dtype = torch.bfloat16
 num_images_per_prompt = 2
+prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype).to(device)
 decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype).to(device)
 prompt = "Anthropomorphic cat dressed as a pilot"