stabilityai
/

stable-cascade

Model card Files Files and versions

dome272 commited on Feb 13, 2024

Commit

90a3d8f

·

verified ·

1 Parent(s): 31a4dd2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -84,7 +84,7 @@ decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-casca
 prompt = "Anthropomorphic cat dressed as a pilot"
 negative_prompt = ""
-with torch.cuda.amp.autocast(dtype=torch.bfloat16):
     prior_output = prior_pipeline(
         prompt=prompt,
         height=1024,

 prompt = "Anthropomorphic cat dressed as a pilot"
 negative_prompt = ""
+with torch.cuda.amp.autocast(dtype=dtype):
     prior_output = prior_pipeline(
         prompt=prompt,
         height=1024,