nicolas-dufour
/

miro

reward-conditioning

Model card Files Files and versions

nicolas-dufour commited on 2 days ago

Commit

2fb12fe

·

verified ·

1 Parent(s): b4f32a2

Upload main weights

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -55,7 +55,7 @@ import torch
 from miro import MiroPipeline
 pipe = MiroPipeline.from_pretrained("nicolas-dufour/miro")
-pipe = pipe.to("cuda", torch.float16)
 prompt = (
     "Photography closeup portrait of an adorable rusty brokendown steampunk "

 from miro import MiroPipeline
 pipe = MiroPipeline.from_pretrained("nicolas-dufour/miro")
+pipe = pipe.to("cuda", torch.bfloat16)
 prompt = (
     "Photography closeup portrait of an adorable rusty brokendown steampunk "