nicolas-dufour
/

miro-ablations

reward-conditioning

Model card Files Files and versions

nicolas-dufour commited on 8 days ago

Commit

2edf273

·

verified ·

1 Parent(s): d9616f4

Upload miro-no-sciscore weights

Files changed (1) hide show

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

miro-no-sciscore/README.md CHANGED Viewed

@@ -57,7 +57,7 @@ from miro import MiroPipeline
 pipe = MiroPipeline.from_pretrained(
     "nicolas-dufour/miro-ablations", variant="miro-no-sciscore",
 )
-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-sciscore",
 )
+pipe = pipe.to("cuda", torch.bfloat16)
 prompt = (
     "Photography closeup portrait of an adorable rusty brokendown steampunk "