nicolas-dufour
/

miro-ablations

reward-conditioning

Model card Files Files and versions

nicolas-dufour commited on May 19

Commit

c121915

·

verified ·

1 Parent(s): 4b67518

Upload miro-only-vqa weights

Files changed (1) hide show

miro-only-vqa/README.md +1 -1

miro-only-vqa/README.md CHANGED Viewed

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