stabilityai
/

stable-cascade-prior

StableCascadePriorPipeline

Model card Files Files and versions

dn6 HF Staff commited on Mar 6, 2024

Commit

03f1620

·

1 Parent(s): 59b5b00

update model card

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -71,7 +71,6 @@ import torch
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 device = "cuda"
-dtype = torch.bfloat16
 num_images_per_prompt = 2
 prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to(device)

 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 device = "cuda"
 num_images_per_prompt = 2
 prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to(device)