Hugging Face's logo

stabilityai
/
stable-cascade

Text-to-Image
Diffusers
Safetensors
StableCascadeDecoderPipeline
Model card Files
xet
 Community
68
multimodalart HF Staff commited on
Commit
d6b6de9
·
verified ·
1 Parent(s): 0a8a23c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -78,7 +78,7 @@ device = "cuda"
78
  dtype = torch.bfloat16
79
  num_images_per_prompt = 2
80
 
81
- prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=dtype).to(device)
82
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=dtype).to(device)
83
 
84
  prompt = "Anthropomorphic cat dressed as a pilot"
 
78
  dtype = torch.bfloat16
79
  num_images_per_prompt = 2
80
 
81
+ prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype).to(device)
82
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=dtype).to(device)
83
 
84
  prompt = "Anthropomorphic cat dressed as a pilot"