nicolas-dufour
/

miro-ablations

reward-conditioning

Model card Files Files and versions

nicolas-dufour commited on May 19

Commit

d9616f4

·

verified ·

1 Parent(s): 4267c3c

Upload miro-no-pickscore weights

Files changed (1) hide show

miro-no-pickscore/README.md +1 -1

miro-no-pickscore/README.md CHANGED Viewed

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

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